1 files changed, 6 insertions, 2 deletions
diff --git a/src/ir/module-utils.cpp b/src/ir/module-utils.cpp
index ce6fadda7..d3ca1c192 100644
--- a/src/ir/module-utils.cpp
+++ b/src/ir/module-utils.cpp
@@ -70,7 +70,9 @@ struct CodeScanner
       counts.note(curr->type);
     } else if (curr->is<ArrayNew>()) {
       counts.note(curr->type);
-    } else if (curr->is<ArrayNewSeg>()) {
+    } else if (curr->is<ArrayNewData>()) {
+      counts.note(curr->type);
+    } else if (curr->is<ArrayNewElem>()) {
       counts.note(curr->type);
     } else if (curr->is<ArrayNewFixed>()) {
       counts.note(curr->type);
@@ -79,7 +81,9 @@ struct CodeScanner
       counts.note(copy->srcRef->type);
     } else if (auto* fill = curr->dynCast<ArrayFill>()) {
       counts.note(fill->ref->type);
-    } else if (auto* init = curr->dynCast<ArrayInit>()) {
+    } else if (auto* init = curr->dynCast<ArrayInitData>()) {
+      counts.note(init->ref->type);
+    } else if (auto* init = curr->dynCast<ArrayInitElem>()) {
       counts.note(init->ref->type);
     } else if (auto* cast = curr->dynCast<RefCast>()) {
       counts.note(cast->type);