diff --git a/main.cc b/main.cc
index c71cae1..55e69f0 100644
--- a/main.cc
+++ b/main.cc
@@ -297,10 +297,11 @@ namespace {
       sqs.push_back(sq);
 
       pos = grid.next_pos(pos + idx);
-      idx = grid.largest_square(pos, n);
 
       // Have we reached the end?  If so success!
       if (pos == grid.end()) { break; }
+
+      idx = grid.largest_square(pos, n);
     }
 
     return {length, sqs};