diff --git a/hta/common/execution_trace.py b/hta/common/execution_trace.py
index 17ab9b2..8a2b8ad 100644
--- a/hta/common/execution_trace.py
+++ b/hta/common/execution_trace.py
@@ -177,12 +177,12 @@ def map_func(node_id):
 
     elif column == "output_shapes":
 
-        def map_func(node_id):
+        def map_func(node_id) -> Any:
             return et.nodes[node_id].output_shapes
 
     elif column == "output_types":
 
-        def map_func(node_id):
+        def map_func(node_id) -> Any:
             return et.nodes[node_id].output_types
 
     elif column == "et_node_name":