diff --git a/python/paddle/fluid/tests/unittests/test_elementwise_mod_op.py b/python/paddle/fluid/tests/unittests/test_elementwise_mod_op.py
index dfe5743f7933e..e186a164c4db2 100644
--- a/python/paddle/fluid/tests/unittests/test_elementwise_mod_op.py
+++ b/python/paddle/fluid/tests/unittests/test_elementwise_mod_op.py
@@ -136,13 +136,13 @@ def test_dygraph(self):
             np.testing.assert_allclose(z_expected, z.numpy(), rtol=1e-05)
 
 
-class TestRemainderInplaceOp(unittest.TestCase):
+class TestRemainderInplaceOp(TestRemainderOp):
 
     def _executed_api(self, x, y, name=None):
         return x.remainder_(y, name)
 
 
-class TestRemainderInplaceBroadcastSuccess(unittest.TestCase):
+class TestRemainderInplaceBroadcastSuccess(TestRemainderOp):
 
     def init_data(self):
         self.x_numpy = np.random.rand(2, 3, 4).astype('float')