diff --git a/run_worker.py b/run_worker.py
index a00406b..5d46800 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, call_callback_ahm
+from temporal.activity import calculate_plant_eaf_activity, execute_simulation_activity, update_contribution_bulk_mappings_activity, update_equipment_for_simulation_activity, call_callback_ahm, acquire_lock, release_lock
 from temporal.workflow import SimulationWorkflow
 
 TEMPORAL_URL = os.environ.get("TEMPORAL_URL", "http://192.168.1.86:7233")
@@ -21,7 +21,9 @@ async def main():
                 execute_simulation_activity,
                 calculate_plant_eaf_activity,
                 update_contribution_bulk_mappings_activity,
-                call_callback_ahm
+                call_callback_ahm,
+                acquire_lock,
+                release_lock
             ],
             max_concurrent_workflow_tasks=50,
             max_concurrent_activities=12