1 files changed, 2 insertions, 4 deletions
diff --git a/src/ir/memory-utils.cpp b/src/ir/memory-utils.cpp
index 9ab0e33fc..49ec25529 100644
--- a/src/ir/memory-utils.cpp
+++ b/src/ir/memory-utils.cpp
@@ -39,10 +39,8 @@ bool flatten(Module& wasm) {
 
     void visitMemoryInit(MemoryInit* curr) { noticesSegmentIdentity = true; }
     void visitDataDrop(DataDrop* curr) { noticesSegmentIdentity = true; }
-    void visitArrayNewSeg(ArrayNewSeg* curr) {
-      if (curr->op == NewData) {
-        noticesSegmentIdentity = true;
-      }
+    void visitArrayNewData(ArrayNewData* curr) {
+      noticesSegmentIdentity = true;
     }
   };