diff --git a/lib/NeuraDialect/Transforms/GenerateCodePass.cpp b/lib/NeuraDialect/Transforms/GenerateCodePass.cpp
index 4bcad118..7e3b5e42 100644
--- a/lib/NeuraDialect/Transforms/GenerateCodePass.cpp
+++ b/lib/NeuraDialect/Transforms/GenerateCodePass.cpp
@@ -923,6 +923,7 @@ struct GenerateCodePass
     func.walk([&](Operation *operation) {
       if (operation == func.getOperation()) return;  // Skips function itself.
       if (isReserve(operation)) return; // Skips reserve nodes entirely (bypass later).
+      if (isa<YieldOp>(operation)) return; // Skips yield nodes entirely (bypass later).
 
       int dfg_id = getDfgId(operation);
       if (dfg_id < 0) {