diff --git a/rvc/train/train.py b/rvc/train/train.py
index d9ecf039e..5cc4f681e 100644
--- a/rvc/train/train.py
+++ b/rvc/train/train.py
@@ -43,7 +43,7 @@
     kl_loss,
 )
 from mel_processing import mel_spectrogram_torch, spec_to_mel_torch
-from process_ckpt import *
+from process_ckpt import save_final, extract_small_model
 
 from rvc.lib.infer_pack import commons