diff --git a/hook_kohya_ss_run.py b/hook_kohya_ss_run.py
index 73a2209..85d995b 100644
--- a/hook_kohya_ss_run.py
+++ b/hook_kohya_ss_run.py
@@ -31,6 +31,9 @@ import numpy as np
 import tempfile
 import safetensors.torch
 
+
+import sys
+sys.path.append(os.path.dirname(__file__))
 try:
     import hook_kohya_ss_utils
 except: