diff --git a/paddleslim/dist/single_distiller.py b/paddleslim/dist/single_distiller.py
index 88d34d3b2b2378681c9e35e10380d8514a9a54aa..b6f9a45e543ea6b1fe64e29eefeaae418510d9d9 100644
--- a/paddleslim/dist/single_distiller.py
+++ b/paddleslim/dist/single_distiller.py
@@ -162,7 +162,7 @@ def soft_label_loss(teacher_var_name,
     return soft_label_loss
 
 
-def self_defined_loss(program, loss_func, **kwargs):
+def loss(program, loss_func, **kwargs):
     """
     Combine variables from student model and teacher model by self defined loss.
     Args: