3 files changed, 8 insertions, 4 deletions
diff --git a/src/s2wasm.h b/src/s2wasm.h
index 5460bf467..1fa624dd0 100644
--- a/src/s2wasm.h
+++ b/src/s2wasm.h
@@ -677,13 +677,13 @@ class S2WasmBuilder {
     auto makeHost = [&](HostOp op) {
       Name assign = getAssign();
       auto curr = allocator.alloc<Host>();
-      curr->op = MemorySize;
+      curr->op = op;
       setOutput(curr, assign);
     };
     auto makeHost1 = [&](HostOp op) {
       Name assign = getAssign();
       auto curr = allocator.alloc<Host>();
-      curr->op = MemorySize;
+      curr->op = op;
       curr->operands.push_back(getInput());
       setOutput(curr, assign);
     };
diff --git a/test/llvm_autogenerated/memory-addr32.wast b/test/llvm_autogenerated/memory-addr32.wast
index 00d1829b2..6e42c7b59 100644
--- a/test/llvm_autogenerated/memory-addr32.wast
+++ b/test/llvm_autogenerated/memory-addr32.wast
@@ -9,7 +9,9 @@
     )
   )
   (func $grow_memory (param $$0 i32)
-    (memory_size)
+    (grow_memory
+      (get_local $$0)
+    )
     (return)
   )
 )
diff --git a/test/llvm_autogenerated/memory-addr64.wast b/test/llvm_autogenerated/memory-addr64.wast
index 06bbf6e11..a7085e3d2 100644
--- a/test/llvm_autogenerated/memory-addr64.wast
+++ b/test/llvm_autogenerated/memory-addr64.wast
@@ -9,7 +9,9 @@
     )
   )
   (func $grow_memory (param $$0 i64)
-    (memory_size)
+    (grow_memory
+      (get_local $$0)
+    )
     (return)
   )
 )