diff --git a/clash_royale/envs/game_engine/arena.py b/clash_royale/envs/game_engine/arena.py
index 2734425..390a463 100644
--- a/clash_royale/envs/game_engine/arena.py
+++ b/clash_royale/envs/game_engine/arena.py
@@ -8,7 +8,7 @@
 
 from typing import TYPE_CHECKING
 
-from clash_royale.envs.game_engine.entities.entity import Entity, EntityCollection
+from clash_royale.envs.game_engine.entities.entity import EntityCollection
 
 if TYPE_CHECKING:
     from clash_royale.envs.game_engine.game_engine import GameEngine
diff --git a/clash_royale/envs/game_engine/logic/attack.py b/clash_royale/envs/game_engine/logic/attack.py
index 221037c..5065cbc 100644
--- a/clash_royale/envs/game_engine/logic/attack.py
+++ b/clash_royale/envs/game_engine/logic/attack.py
@@ -15,6 +15,7 @@
 from clash_royale.envs.game_engine.utils import distance
 
 if TYPE_CHECKING:
+    # Only import for typechecking to prevent circular dependency
     from clash_royale.envs.game_engine.entities.logic_entity import LogicEntity