1 files changed, 2 insertions, 5 deletions
diff --git a/source/slang/ir-missing-return.cpp b/source/slang/ir-missing-return.cpp
index 0396fbce9..c32b71ab6 100644
--- a/source/slang/ir-missing-return.cpp
+++ b/source/slang/ir-missing-return.cpp
@@ -26,12 +26,9 @@ void checkForMissingReturnsRec(
         }
     }
 
-    if( auto parentInst = as<IRParentInst>(inst) )
+    for( auto childInst : inst->getDecorationsAndChildren() )
     {
-        for( auto childInst : parentInst->getChildren() )
-        {
-            checkForMissingReturnsRec(childInst, sink);
-        }
+        checkForMissingReturnsRec(childInst, sink);
     }
 }