diff --git a/prim_wallpaper/pickers/diagonal.py b/prim_wallpaper/pickers/diagonal.py
index 872e19f..6d0d76c 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 += 1
+            row = get_row_num(self.table ** 2 - (i + 1))
         if row % 2 != 0:
             return self.diagonal(i)
         else: