diff --git a/paddleslim/auto_compression/create_compressed_program.py b/paddleslim/auto_compression/create_compressed_program.py
index 60885c4c81b2be3b0bc1e0fb972a691610d22569..7332619a30f807cce3e5cc06de6fc5ade4f98d15 100644
--- a/paddleslim/auto_compression/create_compressed_program.py
+++ b/paddleslim/auto_compression/create_compressed_program.py
@@ -539,8 +539,8 @@ def build_prune_program(executor,
         train_program_info.program = pruned_program
 
     elif strategy.startswith('asp'):
-        from paddle.static import sparsity
-        pruner = sparsity
+        from paddle.incubate import asp
+        pruner = asp
         excluded_params_name = []
         if config['prune_params_name'] is None:
             config['prune_params_name'] = _get_asp_prune_params(