diff --git a/python/paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py
index cdc67cf464752824123b9b3a7b8c793c6c37285c..6996554d427fd366169ddb86cd93f48d724619fc 100644
--- a/python/paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py
+++ b/python/paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py
@@ -38,6 +38,7 @@ class XPUTestSqueeze2Op(XPUOpTestWrapper):
 
         def setUp(self):
             self.op_type = "squeeze2"
+            self.__class__.op_type = "squeeze2"
             self.use_mkldnn = False
             self.init_dtype()
             self.init_test_case()
diff --git a/python/paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py
index 6daa47394122ae2ce0746bc40deee9e3466264b3..e9fc66ca4fcce5c5f8e3caa59a46aac6a25c62e0 100644
--- a/python/paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py
+++ b/python/paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py
@@ -39,6 +39,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper):
 
         def setUp(self):
             self.op_type = "unsqueeze2"
+            self.__class__.op_type = "unsqueeze2"
             self.use_mkldnn = False
             self.init_dtype()
             self.init_test_case()
@@ -117,6 +118,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper):
 
         def setUp(self):
             self.op_type = "unsqueeze2"
+            self.__class__.op_type = "unsqueeze2"
             self.use_mkldnn = False
             self.init_dtype()
             self.init_test_case()
@@ -191,6 +193,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper):
 
         def setUp(self):
             self.op_type = "unsqueeze2"
+            self.__class__.op_type = "unsqueeze2"
             self.use_mkldnn = False
             self.init_test_case()
             self.init_dtype()