diff --git a/src/adf_core_python/core/component/action/extend_action.py b/src/adf_core_python/core/component/action/extend_action.py
index 054f433b..e33f65fd 100644
--- a/src/adf_core_python/core/component/action/extend_action.py
+++ b/src/adf_core_python/core/component/action/extend_action.py
@@ -25,11 +25,11 @@ def __init__(
     module_manager: ModuleManager,
     develop_data: DevelopData,
   ):
-    self.world_info = world_info
-    self.agent_info = agent_info
-    self.scenario_info = scenario_info
-    self.module_manager = module_manager
-    self.develop_data = develop_data
+    self.world_info: WorldInfo = world_info
+    self.agent_info: AgentInfo = agent_info
+    self.scenario_info: ScenarioInfo = scenario_info
+    self.module_manager: ModuleManager = module_manager
+    self.develop_data: DevelopData = develop_data
     self.result: Optional[Action] = None
     self.count_precompute: int = 0
     self.count_resume: int = 0
diff --git a/src/adf_core_python/core/component/centralized/command_executor.py b/src/adf_core_python/core/component/centralized/command_executor.py
index 6b9bd8cb..7c3ac9bf 100644
--- a/src/adf_core_python/core/component/centralized/command_executor.py
+++ b/src/adf_core_python/core/component/centralized/command_executor.py
@@ -29,11 +29,11 @@ def __init__(
     module_manager: ModuleManager,
     develop_data: DevelopData,
   ) -> None:
-    self._agent_info = agent_info
-    self._world_info = world_info
-    self._scenario_info = scenario_info
-    self._module_manager = module_manager
-    self._develop_data = develop_data
+    self._agent_info: AgentInfo = agent_info
+    self._world_info: WorldInfo = world_info
+    self._scenario_info: ScenarioInfo = scenario_info
+    self._module_manager: ModuleManager = module_manager
+    self._develop_data: DevelopData = develop_data
 
     self._result: Optional[Action] = None
 
diff --git a/src/adf_core_python/core/component/centralized/command_picker.py b/src/adf_core_python/core/component/centralized/command_picker.py
index a78c05d8..4f659d75 100644
--- a/src/adf_core_python/core/component/centralized/command_picker.py
+++ b/src/adf_core_python/core/component/centralized/command_picker.py
@@ -28,11 +28,11 @@ def __init__(
     module_manager: ModuleManager,
     develop_data: DevelopData,
   ) -> None:
-    self._agent_info = agent_info
-    self._world_info = world_info
-    self._scenario_info = scenario_info
-    self._module_manager = module_manager
-    self._develop_data = develop_data
+    self._agent_info: AgentInfo = agent_info
+    self._world_info: WorldInfo = world_info
+    self._scenario_info: ScenarioInfo = scenario_info
+    self._module_manager: ModuleManager = module_manager
+    self._develop_data: DevelopData = develop_data
     self._count_precompute: int = 0
     self._count_prepare: int = 0
     self._count_resume: int = 0