1 files changed, 6 insertions, 3 deletions
diff --git a/source/slang/slang-ir.cpp b/source/slang/slang-ir.cpp
index 37841c35f..bc7f7970b 100644
--- a/source/slang/slang-ir.cpp
+++ b/source/slang/slang-ir.cpp
@@ -678,7 +678,7 @@ namespace Slang
         block->insertAtEnd(this);
     }
 
-    void fixUpFuncType(IRFunc* func)
+    void fixUpFuncType(IRFunc* func, IRType* resultType)
     {
         SLANG_ASSERT(func);
 
@@ -699,12 +699,15 @@ namespace Slang
             paramTypes.add(param->getFullType());
         }
 
-        auto resultType = func->getResultType();
-
         auto funcType = builder.getFuncType(paramTypes, resultType);
         builder.setDataType(func, funcType);
     }
 
+    void fixUpFuncType(IRFunc* func)
+    {
+        fixUpFuncType(func, func->getResultType());
+    }
+
     //
 
     bool isTerminatorInst(IROp op)