diff --git a/playground/structs/flow_example.py b/playground/structs/flow_example.py
index f7e1bded..8b9d6325 100644
--- a/playground/structs/flow_example.py
+++ b/playground/structs/flow_example.py
@@ -1,6 +1,8 @@
 from swarms.models import OpenAIChat
 from swarms.structs import Agent
 
+import os
+
 api_key = os.getenv("OPENAI_API_KEY")
 org_id = os.environ.get("OPENAI_ORG_ID")
 
diff --git a/swarms/agents/worker_agent.py b/swarms/agents/worker_agent.py
index cd772c0b..ae1d3d9e 100644
--- a/swarms/agents/worker_agent.py
+++ b/swarms/agents/worker_agent.py
@@ -9,7 +9,7 @@ from langchain_experimental.autonomous_agents import AutoGPT
 
 from swarms.utils.decorators import error_decorator, timing_decorator
 
-load_dotenv()
+import os
 
 api_key = os.getenv("OPENAI_API_KEY")
 org_id = os.getenv("OPENAI_ORG_ID")