diff --git a/launch.py b/launch.py
index bb15e768..33e43fb0 100644
--- a/launch.py
+++ b/launch.py
@@ -1,5 +1,3 @@
-import cuda_malloc
-
 import os
 import sys
 import platform
diff --git a/webui.py b/webui.py
index 4d516a18..5d3f06ef 100644
--- a/webui.py
+++ b/webui.py
@@ -1,3 +1,5 @@
+import cuda_malloc
+
 import gradio as gr
 import random