diff --git a/src/Nncase.Passes/GraphPartition/ExprReConstructor.cs b/src/Nncase.Passes/GraphPartition/ExprReConstructor.cs
index 2de852028..a9b15ae42 100644
--- a/src/Nncase.Passes/GraphPartition/ExprReConstructor.cs
+++ b/src/Nncase.Passes/GraphPartition/ExprReConstructor.cs
@@ -122,23 +122,13 @@ protected override Expr DispatchVisit(Expr expr, Unit context)
         return MarkVisited(expr, base.DispatchVisit(expr, context));
     }
 
-    protected override Expr VisitLeafVar(Var expr, Unit context)
-    {
-        return expr;
-    }
+    protected override Expr VisitLeafFunction(Function expr, Unit context) => expr;
 
-    protected override Expr VisitLeafConst(Const expr, Unit context)
-    {
-        return expr;
-    }
+    protected override Expr VisitLeafVar(Var expr, Unit context) => expr;
 
-    protected override Expr VisitLeafNone(None expr, Unit context)
-    {
-        return expr;
-    }
+    protected override Expr VisitLeafConst(Const expr, Unit context) => expr;
 
-    protected override Expr VisitLeafOp(Op expr, Unit context)
-    {
-        return expr;
-    }
+    protected override Expr VisitLeafNone(None expr, Unit context) => expr;
+
+    protected override Expr VisitLeafOp(Op expr, Unit context) => expr;
 }