diff --git a/app/agent/__init__.py b/app/agent/__init__.py
index 55f2fea9..e6c78ae7 100644
--- a/app/agent/__init__.py
+++ b/app/agent/__init__.py
@@ -13,7 +13,7 @@ from langchain_core.messages import (
 )
 from langgraph.checkpoint.memory import InMemorySaver
 
-from app.agent.callback import StreamingHandler, StreamingHandler
+from app.agent.callback import StreamingHandler
 from app.agent.memory import memory_manager
 from app.agent.middleware.memory import MemoryMiddleware
 from app.agent.middleware.patch_tool_calls import PatchToolCallsMiddleware
diff --git a/app/agent/middleware/memory.py b/app/agent/middleware/memory.py
index 5616314c..d661006d 100644
--- a/app/agent/middleware/memory.py
+++ b/app/agent/middleware/memory.py
@@ -1,7 +1,7 @@
 from collections.abc import Awaitable, Callable
 from typing import Annotated, NotRequired, TypedDict, Dict
 
-from aiopath import AsyncPath
+from aiopathlib import AsyncPath
 from langchain.agents.middleware.types import (
     AgentMiddleware,
     AgentState,