diff --git a/clash_royale/envs/game_engine/entities/entity.py b/clash_royale/envs/game_engine/entities/entity.py
index 03172d2..6eb6954 100644
--- a/clash_royale/envs/game_engine/entities/entity.py
+++ b/clash_royale/envs/game_engine/entities/entity.py
@@ -424,7 +424,7 @@ def start(self) -> None:
 
                 self.start_entity(mod)
 
-    def stop(self):
+    def stop(self) -> None:
         """
         Method used to stop this entityCollection.
 
@@ -473,7 +473,7 @@ def _load_entity(self, entity: Entity) -> None:
 
         entity.collection = self
 
-    def _unload_entity(self, entity: Entity):
+    def _unload_entity(self, entity: Entity) -> None:
         """
         Low-level method for unloading entities from the list.