diff --git a/imperative/python/megengine/jit/tracing.py b/imperative/python/megengine/jit/tracing.py
index b3a1af890523b5b3e1aa811b93c713458468108e..e93b7087b4c8b96ad8caf50eca03912e5015e823 100644
--- a/imperative/python/megengine/jit/tracing.py
+++ b/imperative/python/megengine/jit/tracing.py
@@ -10,6 +10,7 @@ import weakref
 import numpy as np
 
 from ..core._imperative_rt import GraphProfiler
+from ..core._imperative_rt.ops import OprAttr
 from ..core.ops.special import Const
 from ..core.tensor import megbrain_graph as G
 from ..core.tensor.core import OpBase, TensorBase, TensorWrapperBase, apply
@@ -127,8 +128,12 @@ class trace:
         record = self._seq[self._pc]
         op_, ihandles, ohandles = record
         if op != op_:
-            if op.type == "UniformRNG":
-                pass
+            # FIXME: will be removed once better rng implementation is done
+            if isinstance(op, OprAttr) and (
+                op.type in ("UniformRNG", "GaussianRNG") and op.type == op_.type
+            ):
+                if op.param[8:] != op_.param[8:]:
+                    raise TraceMismatchError("op different from last time")
             else:
                 raise TraceMismatchError("op different from last time")
         if len(ihandles) != len(args):