diff --git a/src/Nncase.Passes/Rules/Neutral/FoldBinary.cs b/src/Nncase.Passes/Rules/Neutral/FoldBinary.cs
index c844136f1c..6d15614904 100644
--- a/src/Nncase.Passes/Rules/Neutral/FoldBinary.cs
+++ b/src/Nncase.Passes/Rules/Neutral/FoldBinary.cs
@@ -23,13 +23,14 @@ public sealed partial class FoldNopBinary : IRewriteRule
     /// <inheritdoc/>
     public IPattern Pattern { get; } = IsBinary(
         "binary",
+        "call",
         x => x.BinaryOp is BinaryOp.Add or BinaryOp.Sub or BinaryOp.Mul or BinaryOp.Div or BinaryOp.Mod or BinaryOp.Pow,
         IsWildcard("lhs"),
         IsTensorConst("rhs"));
 
-    private Expr? GetReplace(Binary binary, Expr lhs, TensorConst rhs)
+    private Expr? GetReplace(Binary binary, Call call, Expr lhs, TensorConst rhs)
     {
-        if (lhs.CheckedType is Nncase.IR.AnyType || lhs.CheckedShape == rhs.CheckedShape)
+        if ((lhs.CheckedType is Nncase.IR.AnyType && rhs.CheckedShape.IsScalar) || (lhs.CheckedShape == call.CheckedShape))
         {
             return binary.BinaryOp switch
             {