diff --git a/server/graph_service/routers/retrieve.py b/server/graph_service/routers/retrieve.py
index 333ca0b4..733766e4 100644
--- a/server/graph_service/routers/retrieve.py
+++ b/server/graph_service/routers/retrieve.py
@@ -1,5 +1,4 @@
 from datetime import datetime
-from typing import List, Optional, cast
 
 from fastapi import APIRouter, status
 
@@ -18,9 +17,7 @@ router = APIRouter()
 @router.post('/search', status_code=status.HTTP_200_OK)
 async def search(query: SearchQuery, graphiti: ZepGraphitiDep):
     relevant_edges = await graphiti.search(
-        group_ids=cast(
-            Optional[List[Optional[str]]], query.group_ids
-        ),  # Cast query.group_ids to match the expected type in graphiti.search
+        group_ids=query.group_ids,
         query=query.query,
         num_results=query.max_facts,
     )