diff --git a/workflows/code_implementation_workflow.py b/workflows/code_implementation_workflow.py
index 882a6291..06405ea5 100644
--- a/workflows/code_implementation_workflow.py
+++ b/workflows/code_implementation_workflow.py
@@ -33,6 +33,7 @@
 from workflows.agents.memory_agent_concise import ConciseMemoryAgent
 from config.mcp_tool_definitions import get_mcp_tools
 from utils.llm_utils import get_preferred_llm_class, get_default_models, load_api_config
+from utils.loop_detector import LoopDetector, ProgressTracker
 # DialogueLogger removed - no longer needed
 
 
diff --git a/workflows/code_implementation_workflow_index.py b/workflows/code_implementation_workflow_index.py
index 12d8049b..04b5c26f 100644
--- a/workflows/code_implementation_workflow_index.py
+++ b/workflows/code_implementation_workflow_index.py
@@ -22,6 +22,7 @@
 
 # MCP Agent imports
 from mcp_agent.agents.agent import Agent
+from utils.loop_detector import LoopDetector, ProgressTracker
 
 # Local imports
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))