1 files changed, 5 insertions, 0 deletions
diff --git a/src/passes/Inlining.cpp b/src/passes/Inlining.cpp
index bcab7318f..a44f02426 100644
--- a/src/passes/Inlining.cpp
+++ b/src/passes/Inlining.cpp
@@ -211,6 +211,11 @@ struct Updater : public PostWalker<Updater> {
       handleReturnCall(curr, curr->sig.results);
     }
   }
+  void visitCallRef(CallRef* curr) {
+    if (curr->isReturn) {
+      handleReturnCall(curr, curr->target->type);
+    }
+  }
   void visitLocalGet(LocalGet* curr) {
     curr->index = localMapping[curr->index];
   }