diff --git a/src/cleveragents/templates/inline_jinja_handler.py b/src/cleveragents/templates/inline_jinja_handler.py
index 00ee6c72c..15f40e4fd 100644
--- a/src/cleveragents/templates/inline_jinja_handler.py
+++ b/src/cleveragents/templates/inline_jinja_handler.py
@@ -45,7 +45,7 @@ class InlineJinjaHandler:
 
     def process_yaml_file(
         self, file_path: Path, defer_rendering: bool = True
-    ) -> Any:
+    ) -> dict[str, Any]:
         """
         Process a YAML file containing inline Jinja2 templates.
 
@@ -65,8 +65,8 @@ class InlineJinjaHandler:
         self,
         yaml_content: str,
         defer_rendering: bool = True,
-        context: Optional[dict[str, Any]] = None,
-    ) -> Any:
+        context: Optional[Dict[str, Any]] = None,
+    ) -> dict[str, Any]:
         """
         Process YAML content containing inline Jinja2 templates.
 
@@ -97,7 +97,7 @@ class InlineJinjaHandler:
 
     def _extract_templates(  # pylint: disable=too-many-locals,too-many-branches,too-many-statements,too-many-nested-blocks
         self, yaml_content: str
-    ) -> Any:
+    ) -> dict[str, Any]:
         """
         Extract templates and replace with placeholders.