diff --git a/infer/modules/train/preprocess.py b/infer/modules/train/preprocess.py
index a7cab57..7ff76ee 100644
--- a/infer/modules/train/preprocess.py
+++ b/infer/modules/train/preprocess.py
@@ -18,8 +18,8 @@ import traceback
 
 import librosa
 import numpy as np
-from lib.audio import load_audio
-from lib.slicer2 import Slicer
+from infer.lib.audio import load_audio
+from infer.lib.slicer2 import Slicer
 from scipy.io import wavfile
 
 mutex = multiprocessing.Lock()