diff --git a/streamlit_pages/chat.py b/streamlit_pages/chat.py
index 6800c8e3..af506dad 100644
--- a/streamlit_pages/chat.py
+++ b/streamlit_pages/chat.py
@@ -1,3 +1,4 @@
+from langgraph.types import Command
 import streamlit as st
 import uuid
 import sys
diff --git a/streamlit_ui.py b/streamlit_ui.py
index 51ff9774..7d5694c6 100644
--- a/streamlit_ui.py
+++ b/streamlit_ui.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 from supabase import Client, create_client
 from typing import Literal, TypedDict
-from langgraph.types import Command
 from urllib.parse import urlparse
 from openai import AsyncOpenAI
 from dotenv import load_dotenv