diff --git a/test/autograd/test_autograd_dynamic.py b/test/autograd/test_autograd_dynamic.py
index e8e5d8b626c5f1a66d5f724ddd6dc4a830adf92d..2e6033a6ee937c792cf14ba1f9c4b8a64eb1ce60 100644
--- a/test/autograd/test_autograd_dynamic.py
+++ b/test/autograd/test_autograd_dynamic.py
@@ -630,7 +630,7 @@ class TestHessianBatchFirst(unittest.TestCase):
 
     def func_ndim_bigger_than_2(self):
         def func(x):
-            return (x * x).sum()
+            return (x * x).sum([1, 2, 3])
 
         with self.assertRaises(ValueError):
             x = paddle.ones([3, 3, 3, 3])