diff --git a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEFuseFillIntoForall.cpp b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEFuseFillIntoForall.cpp
index 3a515e08f..71e416021 100644
--- a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEFuseFillIntoForall.cpp
+++ b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEFuseFillIntoForall.cpp
@@ -83,9 +83,7 @@ void AMDAIEFuseFillIntoForallPass::runOnOperation() {
   rewriter.replaceUsesWithIf(
       bbArg, fusedFill.getResult(0), [&](OpOperand &operand) {
         Operation *owner = operand.getOwner();
-        if (owner == fusedFill) {
-          return false;
-        } else if (isa<tensor::ParallelInsertSliceOp>(owner)) {
+        if (owner == fusedFill || isa<tensor::ParallelInsertSliceOp>(owner)) {
           return false;
         } else {
           return true;