diff --git a/lang_agent/base.py b/lang_agent/base.py
index 31e3096..479744a 100644
--- a/lang_agent/base.py
+++ b/lang_agent/base.py
@@ -5,8 +5,7 @@ from io import BytesIO
 import matplotlib.pyplot as plt
 from loguru import logger
 
-if TYPE_CHECKING:
-    from langgraph.graph.state import CompiledStateGraph
+from langgraph.graph.state import CompiledStateGraph
 
 
 class LangToolBase(ABC):