diff --git a/src/agent/graph.py b/src/agent/graph.py
index d160e9d..2f9d72c 100644
--- a/src/agent/graph.py
+++ b/src/agent/graph.py
@@ -1,5 +1,8 @@
 """Module for defining the agent's workflow graph and human interaction nodes."""
 
+from __future__ import annotations
+
+from dataclasses import dataclass
 from typing import Any, Dict
 
 from langchain_core.runnables import RunnableConfig
@@ -12,7 +15,13 @@
 )
 from langgraph.types import interrupt
 
-from agent.state import State
+
+@dataclass
+class State:
+    """Class representing the state of the agent interaction."""
+
+    interrupt_response: str = "example"
+
 
 
 async def human_node(state: State, config: RunnableConfig) -> Dict[str, Any]:
diff --git a/src/agent/state.py b/src/agent/state.py
deleted file mode 100644
index 76e43a4..0000000
--- a/src/agent/state.py
+++ /dev/null
@@ -1,12 +0,0 @@
-"""State module for managing agent state."""
-
-from __future__ import annotations
-
-from dataclasses import dataclass
-
-
-@dataclass
-class State:
-    """Class representing the state of the agent interaction."""
-
-    interrupt_response: str = "example"