From fa6624ea749b82b190b99ca9f64a7f85efc017f4 Mon Sep 17 00:00:00 2001
From: Shaz Qadeer <shaz.qadeer@gmail.com>
Date: Thu, 25 Apr 2024 13:04:59 -0700
Subject: [PATCH] [Civl] Fixed bug in typechecking of yielding loops (#877)

Co-authored-by: Shaz Qadeer <shaz@meta.com>
---
 Source/Core/AST/Absy.cs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Source/Core/AST/Absy.cs b/Source/Core/AST/Absy.cs
index f04a317bb..b82fd2dfe 100644
--- a/Source/Core/AST/Absy.cs
+++ b/Source/Core/AST/Absy.cs
@@ -4121,9 +4121,9 @@ public override void Typecheck(TypecheckingContext tc)
     private void TypecheckLoopAnnotations(TypecheckingContext tc, Graph<Block> graph)
     {
       var yieldingProc = (YieldProcedureDecl)Proc;
-      var yieldingLayer = yieldingProc.Layer;
       foreach (var header in graph.Headers)
       {
+        var yieldingLayer = yieldingProc.Layer;
         var yieldCmd = (PredicateCmd)header.Cmds.FirstOrDefault(cmd =>
           cmd is PredicateCmd predCmd && predCmd.HasAttribute(CivlAttributes.YIELDS));
         if (yieldCmd == null)