1 files changed, 3 insertions, 5 deletions
diff --git a/src/passes/ReorderFunctions.cpp b/src/passes/ReorderFunctions.cpp
index 0c95101a5..66b8275ef 100644
--- a/src/passes/ReorderFunctions.cpp
+++ b/src/passes/ReorderFunctions.cpp
@@ -29,6 +29,7 @@
 
 #include <memory>
 
+#include <ir/element-utils.h>
 #include <pass.h>
 #include <wasm.h>
 
@@ -70,11 +71,8 @@ struct ReorderFunctions : public Pass {
     for (auto& curr : module->exports) {
       counts[curr->value]++;
     }
-    for (auto& segment : module->elementSegments) {
-      for (auto& curr : segment->data) {
-        counts[curr]++;
-      }
-    }
+    ElementUtils::iterAllElementFunctionNames(
+      module, [&](Name& name) { counts[name]++; });
     // sort
     std::sort(module->functions.begin(),
               module->functions.end(),