diff --git a/python/paddle/tests/test_model.py b/python/paddle/tests/test_model.py
index 69a888de9c61fd649239d196f70a5b3e57039863..c5a0e820fe793f995c8fc4b8a9b86a6ec8b52a87 100644
--- a/python/paddle/tests/test_model.py
+++ b/python/paddle/tests/test_model.py
@@ -616,7 +616,7 @@ class TestRaiseError(unittest.TestCase):
 
     def test_export_deploy_model_without_inputs_and_run_in_dygraph(self):
         paddle.disable_static()
-        net = MyModel(classifier_activation=None)
+        net = MyModel()
         save_dir = tempfile.mkdtemp()
         if not os.path.exists(save_dir):
             os.makedirs(save_dir)