diff --git a/prim_wallpaper/pickers/diagonal.py b/prim_wallpaper/pickers/diagonal.py
index 178bb0e..5fa5d45 100644
--- a/prim_wallpaper/pickers/diagonal.py
+++ b/prim_wallpaper/pickers/diagonal.py
@@ -30,7 +30,8 @@ class DiagonalCounter:
 
     def diagonal_then_reversed(self, i):
         row = get_row_num(i)
-        if row > self.table_size:
+
+        if row > self.table_size - 1:
             row = get_row_num(self.table_size ** 2 - (i + 1))
         if row % 2 != 0:
             return self.diagonal(i)
@@ -49,8 +50,6 @@ if __name__ == "__main__":
         indices = map(func, range(d.table_size ** 2))
         for i, index in enumerate(indices):
             x, y = index
-            if i == 26 and func == d.diagonal_then_reversed:
-                breakpoint()
             output[y][x] = i
 
         with open(func.__name__ + ".csv", "w", newline="") as file: