1 files changed, 2 insertions, 5 deletions
diff --git a/src/passes/RemoveUnusedModuleElements.cpp b/src/passes/RemoveUnusedModuleElements.cpp
index 21cbc5e5b..3795022c2 100644
--- a/src/passes/RemoveUnusedModuleElements.cpp
+++ b/src/passes/RemoveUnusedModuleElements.cpp
@@ -111,11 +111,8 @@ struct ReachabilityAnalyzer : public PostWalker<ReachabilityAnalyzer> {
   void visitDataDrop(DataDrop* curr) { usesMemory = true; }
   void visitMemoryCopy(MemoryCopy* curr) { usesMemory = true; }
   void visitMemoryFill(MemoryFill* curr) { usesMemory = true; }
-  void visitHost(Host* curr) {
-    if (curr->op == MemorySize || curr->op == MemoryGrow) {
-      usesMemory = true;
-    }
-  }
+  void visitMemorySize(MemorySize* curr) { usesMemory = true; }
+  void visitMemoryGrow(MemoryGrow* curr) { usesMemory = true; }
   void visitRefFunc(RefFunc* curr) {
     if (reachable.count(
           ModuleElement(ModuleElementKind::Function, curr->func)) == 0) {