1 files changed, 2 insertions, 3 deletions
diff --git a/Source/Concurrency/TypeCheck.cs b/Source/Concurrency/TypeCheck.cs
index 5fd67978..cd636946 100644
--- a/Source/Concurrency/TypeCheck.cs
+++ b/Source/Concurrency/TypeCheck.cs
@@ -269,10 +269,9 @@ namespace Microsoft.Boogie
 
         public void TypeCheck()
         {
-            foreach (Declaration decl in program.TopLevelDeclarations)
+            foreach (var proc in program.Procedures)
             {
-                Procedure proc = decl as Procedure;
-                if (proc == null || !QKeyValue.FindBoolAttribute(proc.Attributes, "yields")) continue;
+                if (!QKeyValue.FindBoolAttribute(proc.Attributes, "yields")) continue;
 
                 int phaseNum = int.MaxValue;
                 int availableUptoPhaseNum = int.MaxValue;