diff --git a/backend/src/agent/state.py b/backend/src/agent/state.py
index d5ad4dcd8..10f16fbc9 100644
--- a/backend/src/agent/state.py
+++ b/backend/src/agent/state.py
@@ -40,7 +40,7 @@ class QueryGenerationState(TypedDict):
 
 class WebSearchState(TypedDict):
     search_query: str
-    id: str
+    id: int
 
 
 @dataclass(kw_only=True)