diff --git a/demo/gradio/computer_use_demo/agent/llm_utils/omniparserclient.py b/demo/gradio/computer_use_demo/agent/llm_utils/omniparserclient.py
index 7b44069..010e82a 100644
--- a/demo/gradio/computer_use_demo/agent/llm_utils/omniparserclient.py
+++ b/demo/gradio/computer_use_demo/agent/llm_utils/omniparserclient.py
@@ -2,7 +2,7 @@ import requests
 import base64
 from pathlib import Path
 from computer_use_demo.tools.screen_capture import get_screenshot
-from computer_use_demo.gui_agent.llm_utils.oai import encode_image
+from computer_use_demo.agent.llm_utils.oai import encode_image
 
 OUTPUT_DIR = "./tmp/outputs"
 
diff --git a/demo/gradio/computer_use_demo/agent/vlm_agent.py b/demo/gradio/computer_use_demo/agent/vlm_agent.py
index 0ce905a..58e243e 100644
--- a/demo/gradio/computer_use_demo/agent/vlm_agent.py
+++ b/demo/gradio/computer_use_demo/agent/vlm_agent.py
@@ -10,7 +10,7 @@ from anthropic import APIResponse
 from anthropic.types import ToolResultBlockParam
 from anthropic.types.beta import BetaMessage, BetaTextBlock, BetaToolUseBlock, BetaMessageParam, BetaUsage
 
-from computer_use_demo.gui_agent.llm_utils.oai import run_oai_interleaved
+from computer_use_demo.agent.llm_utils.oai import run_oai_interleaved
 from computer_use_demo.colorful_text import colorful_text_vlm
 import time
 import re