1 files changed, 19 insertions, 3 deletions
diff --git a/src/wasm.h b/src/wasm.h
index c1ef75ea9..75d6a174c 100644
--- a/src/wasm.h
+++ b/src/wasm.h
@@ -1609,10 +1609,26 @@ public:
   }
   // TODO: remove* for other elements
 
-  void updateFunctionsMap() {
+  void updateMaps() {
     functionsMap.clear();
-    for (auto& func : functions) {
-      functionsMap[func->name] = func.get();
+    for (auto& curr : functions) {
+      functionsMap[curr->name] = curr.get();
+    }
+    functionTypesMap.clear();
+    for (auto& curr : functionTypes) {
+      functionTypesMap[curr->name] = curr.get();
+    }
+    importsMap.clear();
+    for (auto& curr : imports) {
+      importsMap[curr->name] = curr.get();
+    }
+    exportsMap.clear();
+    for (auto& curr : exports) {
+      exportsMap[curr->name] = curr.get();
+    }
+    globalsMap.clear();
+    for (auto& curr : globals) {
+      globalsMap[curr->name] = curr.get();
     }
   }
 };