diff --git a/python/paddle/fluid/tests/unittests/test_imperative_save_load_v2.py b/python/paddle/fluid/tests/unittests/test_imperative_save_load_v2.py
index e81d1c8610f6bebffadf930b67dc14a4a418ef05..3ccd1dbda3a443d50e43ba498cb3d5b529318c32 100644
--- a/python/paddle/fluid/tests/unittests/test_imperative_save_load_v2.py
+++ b/python/paddle/fluid/tests/unittests/test_imperative_save_load_v2.py
@@ -297,6 +297,7 @@ class TestDygraphPtbRnn(unittest.TestCase):
             paddle.save(self.state_dict, "./test_dy_v2")
 
     def testLoadAndSetVarBase(self):
+        self.setUp()
         seed = 90
         hidden_size = 10
         vocab_size = 1000