diff --git a/paddleslim/dist/single_distiller.py b/paddleslim/dist/single_distiller.py
index 6d79ad676..8b20706aa 100644
--- a/paddleslim/dist/single_distiller.py
+++ b/paddleslim/dist/single_distiller.py
@@ -14,7 +14,6 @@
 
 import numpy as np
 import paddle
-from ..common.recover_program import _recover_outputs_attr
 from paddleslim.core import GraphWrapper
 import paddle.nn.functional as F
 
@@ -77,9 +76,7 @@ def merge(teacher_program,
         scope = paddle.static.global_scope()
     if teacher_scope == None:
         teacher_scope = scope
-
     teacher_program = teacher_program.clone(for_test=True)
-    teacher_program = _recover_outputs_attr(teacher_program)
 
     is_same_model = True
     if len(student_program.blocks) == len(teacher_program.blocks):