diff --git a/laborious/worker/worker.py b/laborious/worker/worker.py
index 50d00cf..33384b0 100644
--- a/laborious/worker/worker.py
+++ b/laborious/worker/worker.py
@@ -43,7 +43,6 @@ Poller Configuration:
 - POLLER_INITIAL: Initial number of pollers (default: 2)
 """
 
-import os
 
 from temporalio import client, workflow
 from temporalio.runtime import PrometheusConfig, Runtime, TelemetryConfig
@@ -80,6 +79,7 @@ with workflow.unsafe.imports_passed_through():
         FormatAndExportPrediction,
     )
     from laborious.workflows.sub_workflows.prediction_process import PredictionProcess
+    import os
 
 POD_ID = os.getenv('POD_ID')
 SDK_METRICS_PORT = int(os.getenv('HTTP_SDK_METRICS_PORT', '9091'))