diff --git a/tests/test_soft_label_loss.py b/tests/test_soft_label_loss.py
index 64544aa67096b4195ca9cbd5e6218f099afe3436..2e0bf8c8b0d62f726608bde352443cffa767023f 100644
--- a/tests/test_soft_label_loss.py
+++ b/tests/test_soft_label_loss.py
@@ -54,9 +54,12 @@ class TestSoftLabelLoss(StaticCase):
         for block in paddle.static.default_main_program().blocks:
             for op in block.ops:
                 loss_ops.append(op.type)
+        print(f"ret: {set(loss_ops).difference(set(merged_ops))}")
         self.assertTrue(set(merged_ops).difference(set(loss_ops)) == set())
-        self.assertTrue({'cross_entropy', 'softmax', 'reduce_mean'}.issubset(
-            set(loss_ops).difference(set(merged_ops))))
+
+        self.assertTrue({
+            'softmax_with_cross_entropy', 'softmax', 'reduce_mean'
+        }.issubset(set(loss_ops).difference(set(merged_ops))))
 
 
 if __name__ == '__main__':