diff --git a/examples/agent_ui_examples.py b/examples/agent_ui_examples.py
index d5a37119..97f54856 100644
--- a/examples/agent_ui_examples.py
+++ b/examples/agent_ui_examples.py
@@ -13,7 +13,7 @@ from utils import load_dotenv_files
 load_dotenv_files()
 
 # Import the create_gradio_ui function
-from agent.ui.gradio.app import create_gradio_ui
+from agent.ui.gradio.ui_components import create_gradio_ui
 
 if __name__ == "__main__":
     print("Launching Computer-Use Agent Gradio UI with advanced features...")
diff --git a/notebooks/agent_nb.ipynb b/notebooks/agent_nb.ipynb
index 4c39c204..61e7288a 100644
--- a/notebooks/agent_nb.ipynb
+++ b/notebooks/agent_nb.ipynb
@@ -379,7 +379,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from agent.ui.gradio.app import create_gradio_ui\n",
+    "from agent.ui.gradio.ui_components import create_gradio_ui\n",
     "\n",
     "app = create_gradio_ui()\n",
     "app.launch(share=False)"
diff --git a/scripts/playground.sh b/scripts/playground.sh
index 39710e4c..0cde5a25 100755
--- a/scripts/playground.sh
+++ b/scripts/playground.sh
@@ -257,7 +257,7 @@ from pathlib import Path
 from dotenv import load_dotenv
 from computer import Computer
 from agent import ComputerAgent, LLM, AgentLoop, LLMProvider
-from agent.ui.gradio.app import create_gradio_ui
+from agent.ui.gradio.ui_components import create_gradio_ui
 
 # Load environment variables from .env.local
 load_dotenv(Path(__file__).parent / ".env.local")
@@ -292,7 +292,7 @@ from pathlib import Path
 from dotenv import load_dotenv
 from computer import Computer
 from agent import ComputerAgent, LLM, AgentLoop, LLMProvider
-from agent.ui.gradio.app import create_gradio_ui
+from agent.ui.gradio.ui_components import create_gradio_ui
 
 # Load environment variables from .env.local
 load_dotenv(Path(__file__).parent / ".env.local")