diff --git a/launch.py b/launch.py
index 33e43fb..bb15e76 100644
--- a/launch.py
+++ b/launch.py
@@ -1,3 +1,5 @@
+import cuda_malloc
+
 import os
 import sys
 import platform
diff --git a/modules/core.py b/modules/core.py
index d731ae7..799314c 100644
--- a/modules/core.py
+++ b/modules/core.py
@@ -1,5 +1,3 @@
-import cuda_malloc
-
 import os
 import random
 import cv2