diff --git a/letta/agent.py b/letta/agent.py
index bff3e233c3..82958acda5 100644
--- a/letta/agent.py
+++ b/letta/agent.py
@@ -33,7 +33,6 @@
 from letta.schemas.block import BlockUpdate
 from letta.schemas.embedding_config import EmbeddingConfig
 from letta.schemas.enums import MessageRole
-from letta.schemas.letta_message import FunctionReturn
 from letta.schemas.memory import ContextWindowOverview, Memory
 from letta.schemas.message import Message, MessageUpdate
 from letta.schemas.openai.chat_completion_request import (
diff --git a/tests/integration_test_agent_tool_graph.py b/tests/integration_test_agent_tool_graph.py
index 6f14740a72..746cc59479 100644
--- a/tests/integration_test_agent_tool_graph.py
+++ b/tests/integration_test_agent_tool_graph.py
@@ -3,7 +3,6 @@
 
 import pytest
 from letta import create_client
-from letta.schemas.letta_message import FunctionCallMessage
 from letta.schemas.tool_rule import (
     ChildToolRule,
     ConditionalToolRule,