diff --git a/iz_helpers/ui.py b/iz_helpers/ui.py
index 2607286..0c6062c 100644
--- a/iz_helpers/ui.py
+++ b/iz_helpers/ui.py
@@ -1,7 +1,7 @@
 import gradio as gr
 from .run import create_zoom
 import modules.shared as shared
-from webui import wrap_gradio_gpu_call
+from modules.call_queue import wrap_gradio_gpu_call
 from modules.ui import create_output_panel
 
 from .static_variables import (