diff --git a/infer/modules/train/extract/extract_f0_print.py b/infer/modules/train/extract/extract_f0_print.py
index d97c766..a3e9c9b 100644
--- a/infer/modules/train/extract/extract_f0_print.py
+++ b/infer/modules/train/extract/extract_f0_print.py
@@ -10,7 +10,7 @@ import logging
 
 import numpy as np
 import pyworld
-from lib.audio import load_audio
+from infer.lib.audio import load_audio
 
 logging.getLogger("numba").setLevel(logging.WARNING)
 from multiprocessing import Process
@@ -82,7 +82,7 @@ class FeatureInput(object):
             f0 = pyworld.stonemask(x.astype(np.double), f0, t, self.fs)
         elif f0_method == "rmvpe":
             if hasattr(self, "model_rmvpe") == False:
-                from lib.rmvpe import RMVPE
+                from infer.lib.rmvpe import RMVPE
 
                 print("loading rmvpe model")
                 self.model_rmvpe = RMVPE(
diff --git a/infer/modules/train/extract/extract_f0_rmvpe.py b/infer/modules/train/extract/extract_f0_rmvpe.py
index 0bb2a3e..98a91e7 100644
--- a/infer/modules/train/extract/extract_f0_rmvpe.py
+++ b/infer/modules/train/extract/extract_f0_rmvpe.py
@@ -10,7 +10,7 @@ import logging
 
 import numpy as np
 import pyworld
-from lib.audio import load_audio
+from infer.lib.audio import load_audio
 
 logging.getLogger("numba").setLevel(logging.WARNING)
 
@@ -45,7 +45,7 @@ class FeatureInput(object):
         # p_len = x.shape[0] // self.hop
         if f0_method == "rmvpe":
             if hasattr(self, "model_rmvpe") == False:
-                from lib.rmvpe import RMVPE
+                from infer.lib.rmvpe import RMVPE
 
                 print("loading rmvpe model")
                 self.model_rmvpe = RMVPE(
diff --git a/infer/modules/train/extract/extract_f0_rmvpe_dml.py b/infer/modules/train/extract/extract_f0_rmvpe_dml.py
index 1616e72..2d5c7f4 100644
--- a/infer/modules/train/extract/extract_f0_rmvpe_dml.py
+++ b/infer/modules/train/extract/extract_f0_rmvpe_dml.py
@@ -10,7 +10,7 @@ import logging
 
 import numpy as np
 import pyworld
-from lib.audio import load_audio
+from infer.lib.audio import load_audio
 
 logging.getLogger("numba").setLevel(logging.WARNING)
 
@@ -43,7 +43,7 @@ class FeatureInput(object):
         # p_len = x.shape[0] // self.hop
         if f0_method == "rmvpe":
             if hasattr(self, "model_rmvpe") == False:
-                from lib.rmvpe import RMVPE
+                from infer.lib.rmvpe import RMVPE
 
                 print("loading rmvpe model")
                 self.model_rmvpe = RMVPE(