diff --git a/official/cv/resnet/golden_stick/pruner/scop/train.py b/official/cv/resnet/golden_stick/pruner/scop/train.py
index 9db5ed87d06873ee1cd539aed43cee5a43364156..107189623c122932bda92745bc968a45b143b6d3 100644
--- a/official/cv/resnet/golden_stick/pruner/scop/train.py
+++ b/official/cv/resnet/golden_stick/pruner/scop/train.py
@@ -25,7 +25,8 @@ from mindspore.communication.management import init, get_rank
 from mindspore.parallel import set_algo_parameters
 from mindspore.train.loss_scale_manager import FixedLossScaleManager
 from mindspore.nn.loss import SoftmaxCrossEntropyWithLogits
-from mindspore_gs import PrunerKfCompressAlgo, PrunerFtCompressAlgo, KfConv2d, MaskedConv2dbn
+from mindspore_gs import PrunerKfCompressAlgo, PrunerFtCompressAlgo
+from mindspore_gs.pruner.scop.scop_pruner import KfConv2d, MaskedConv2dbn
 from src.lr_generator import get_lr
 from src.CrossEntropySmooth import CrossEntropySmooth
 from src.resnet import conv_variance_scaling_initializer