diff --git a/src/assignments/assignment10/small_exercises_grade.rs b/src/assignments/assignment10/small_exercises_grade.rs
index 8cbc96f..595a708 100644
--- a/src/assignments/assignment10/small_exercises_grade.rs
+++ b/src/assignments/assignment10/small_exercises_grade.rs
@@ -340,12 +340,17 @@ mod test {
             (68, 285, 293),
         ];
 
-        for (i, t) in pythagorean().enumerate().take(1000) {
-            if i < pythagoreans.len() {
-                assert_eq!(pythagoreans[i], t)
-            }
-            let (a, b, c) = t;
+        let mut pythagorean_iter = pythagorean();
+        for i in 0..1000 {
+            let t = pythagorean_iter.next();
+            assert!(t.is_some());
+
+            let (a, b, c) = t.unwrap();
             assert_eq!(a * a + b * b, c * c);
+
+            if i < pythagoreans.len() {
+                assert_eq!(pythagoreans[i], (a, b, c))
+            }
         }
     }
 }