diff --git a/pystencils/simp/assignment_collection.py b/pystencils/simp/assignment_collection.py
index fa437a5d0fbe49ed04cabeaaf3e22d335b939819..1e9a4c47e64e5a5683076a6fa912bd7298aa6c69 100644
--- a/pystencils/simp/assignment_collection.py
+++ b/pystencils/simp/assignment_collection.py
@@ -347,7 +347,7 @@ class AssignmentCollection:
         return result
 
     def __iter__(self):
-        return self.main_assignments.__iter__()
+        return self.all_assignments.__iter__()
 
     @property
     def main_assignments_dict(self):