diff --git a/slim/prune/eval_prune.py b/slim/prune/eval_prune.py
index 940adce015b9e703535755cec06f57d75acbb051..b8275d03475b8fea67d73682b54a38172fbc25e2 100644
--- a/slim/prune/eval_prune.py
+++ b/slim/prune/eval_prune.py
@@ -43,7 +43,7 @@ from models.model_builder import ModelPhase
 from reader import SegDataset
 from metrics import ConfusionMatrix
 
-from paddleslim.prune.io import *
+from paddleslim.prune import load_model
 
 def parse_args():
     parser = argparse.ArgumentParser(description='PaddleSeg model evalution')