diff --git a/run_worker.py b/run_worker.py
index 678c899..a00406b 100644
--- a/run_worker.py
+++ b/run_worker.py
@@ -3,7 +3,7 @@ import os
 from temporalio.client import Client
 from temporalio.worker import Worker
 
-from temporal.activity import calculate_plant_eaf_activity, execute_simulation_activity, update_contribution_bulk_mappings_activity, update_equipment_for_simulation_activity
+from temporal.activity import calculate_plant_eaf_activity, execute_simulation_activity, update_contribution_bulk_mappings_activity, update_equipment_for_simulation_activity, call_callback_ahm
 from temporal.workflow import SimulationWorkflow
 
 TEMPORAL_URL = os.environ.get("TEMPORAL_URL", "http://192.168.1.86:7233")
@@ -20,7 +20,8 @@ async def main():
                 update_equipment_for_simulation_activity,
                 execute_simulation_activity,
                 calculate_plant_eaf_activity,
-                update_contribution_bulk_mappings_activity
+                update_contribution_bulk_mappings_activity,
+                call_callback_ahm
             ],
             max_concurrent_workflow_tasks=50,
             max_concurrent_activities=12
diff --git a/temporal/workflow.py b/temporal/workflow.py
index f40e74e..fd1717a 100644
--- a/temporal/workflow.py
+++ b/temporal/workflow.py
@@ -1,7 +1,8 @@
 from datetime import timedelta
 from temporalio import workflow
 
-from temporal.activity import calculate_plant_eaf_activity, execute_simulation_activity, update_contribution_bulk_mappings_activity, update_equipment_for_simulation_activity,call_callback_ahm
+with workflow.unsafe.imports_passed_through():
+    from temporal.activity import calculate_plant_eaf_activity, execute_simulation_activity, update_contribution_bulk_mappings_activity, update_equipment_for_simulation_activity,call_callback_ahm