diff --git a/pentestagent/agents/base_agent.py b/pentestagent/agents/base_agent.py
index 44228d3..539511c 100644
--- a/pentestagent/agents/base_agent.py
+++ b/pentestagent/agents/base_agent.py
@@ -5,7 +5,7 @@ from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Any, AsyncIterator, List, Optional
 
 from ..config.constants import AGENT_MAX_ITERATIONS
-from ..workspaces.manager import TargetManager, WorkspaceManager
+from ..workspaces.manager import WorkspaceManager
 from ..workspaces import validation
 from .state import AgentState, AgentStateManager