diff --git a/tests/ut/python/pipeline/parse/test_parse.py b/tests/ut/python/pipeline/parse/test_parse.py
index db0fa42162d..3a8e182b847 100644
--- a/tests/ut/python/pipeline/parse/test_parse.py
+++ b/tests/ut/python/pipeline/parse/test_parse.py
@@ -277,7 +277,7 @@ class AssignWhenInsertGrad(nn.Cell):
         out = self.getG(out)
         return out
 
-grad_all = C.GradOperation('get_all', get_all=True)
+grad_all = C.GradOperation(get_all=True)
 
 class GradNet(nn.Cell):
     def __init__(self, net):