diff --git a/onnxscript/rewriter/_rewrite_rule.py b/onnxscript/rewriter/_rewrite_rule.py
index 3e8b9e7faf..f22374b753 100644
--- a/onnxscript/rewriter/_rewrite_rule.py
+++ b/onnxscript/rewriter/_rewrite_rule.py
@@ -257,7 +257,7 @@ class RewriteRuleClassBase(abc.ABC):
 
     Example::
 
-        class TransposeIdentity(RewriteRuleAsClass):
+        class TransposeIdentity(RewriteRuleClassBase):
             def pattern(cls, op, x, perm):
                 return op.Transpose(x, perm=perm)