diff --git a/mindspore/ccsrc/frontend/optimizer/ad/kprim.cc b/mindspore/ccsrc/frontend/optimizer/ad/kprim.cc
index d9e0f88bb5b..3026e277ac4 100644
--- a/mindspore/ccsrc/frontend/optimizer/ad/kprim.cc
+++ b/mindspore/ccsrc/frontend/optimizer/ad/kprim.cc
@@ -357,9 +357,9 @@ AnfNodePtr KPrim::BuildOutput(const FuncGraphPtr &bprop_fg, const FuncGraphPtr &
       } else if (HasAbstractIOMonad(primal_node)) {
         extra_node = NewValueNode(kIOMonad);
       } else {
-        MS_LOG(EXCEPTION) << "Function: " << current_primal_fg->ToString()
-                          << ", has extra parameter which is not UMoand or IOMonad, but: "
-                          << primal_node->DebugString();
+        MS_EXCEPTION(TypeError) << "Function: " << current_primal_fg->ToString()
+                                << ", has extra parameter which is not UMoand or IOMonad, but: "
+                                << primal_node->DebugString();
       }
       extra_args.push_back(extra_node);
       MS_LOG(DEBUG) << "Insert to bprop_fg for node: " << primal_node->DebugString();
diff --git a/tests/ut/python/pipeline/infer/test_auto_monad.py b/tests/ut/python/pipeline/infer/test_auto_monad.py
index 1ae95718441..a71f340f36c 100644
--- a/tests/ut/python/pipeline/infer/test_auto_monad.py
+++ b/tests/ut/python/pipeline/infer/test_auto_monad.py
@@ -181,7 +181,7 @@ def test_user_defined_bad_bprop():
     net = GradNet(user_defined_net)
     x = Tensor(np.array([2.0], np.float32))
     y = Tensor(np.array([3.0], np.float32))
-    with pytest.raises(RuntimeError):
+    with pytest.raises(TypeError):
         net(x, y)