diff --git a/imperative/python/megengine/jit/tracing.py b/imperative/python/megengine/jit/tracing.py
index 2c116f2928414785285c14785b044834ca9c95b4..baa40c3462d28c9a6908554523212b7cb9fa676b 100644
--- a/imperative/python/megengine/jit/tracing.py
+++ b/imperative/python/megengine/jit/tracing.py
@@ -155,9 +155,7 @@ class trace:
                             "last time, got an internal tensor this time"
                         )
                     if x._handle != info.bound_data._handle:
-                        if not np.array_equal(
-                            x.numpy(), info.bound_data.numpy(), equal_nan=True
-                        ):
+                        if not np.array_equal(x.numpy(), info.bound_data.numpy()):
                             raise TraceMismatchError(
                                 "const capture violated: got "
                                 "a different tensor this time"