diff --git a/prim_wallpaper/pickers/diagonal.py b/prim_wallpaper/pickers/diagonal.py
index 6d0d76c..178bb0e 100644
--- a/prim_wallpaper/pickers/diagonal.py
+++ b/prim_wallpaper/pickers/diagonal.py
@@ -31,7 +31,7 @@ class DiagonalCounter:
     def diagonal_then_reversed(self, i):
         row = get_row_num(i)
         if row > self.table_size:
-            row = get_row_num(self.table ** 2 - (i + 1))
+            row = get_row_num(self.table_size ** 2 - (i + 1))
         if row % 2 != 0:
             return self.diagonal(i)
         else:
@@ -49,6 +49,8 @@ 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: