diff --git a/python/tvm/relay/ir_pass.py b/python/tvm/relay/ir_pass.py
index 356e746e352ee..38fa26645e17f 100644
--- a/python/tvm/relay/ir_pass.py
+++ b/python/tvm/relay/ir_pass.py
@@ -19,6 +19,7 @@ def post_order_visit(expr, fvisit):
     ----------
     expr : tvm.relay.Expr
         The input expression.
+
     fvisit : function
         The visitor function to be applied.
     """
@@ -35,7 +36,6 @@ def infer_type(expr, mod=None):
     mod: Optional[tvm.relay.Module]
         The global module.
 
-
     Returns
     -------
     checked_expr : tvm.relay.Expr
@@ -368,6 +368,9 @@ def to_anf(expr, mod=None):
     expr : tvm.relay.Expr
         The input expression
 
+    mod: Optional[tvm.relay.Module]
+        The global module.
+
     Returns
     -------
     expr: tvm.relay.Expr