From 71019726d43504314a0fc5f54487609a855460c8 Mon Sep 17 00:00:00 2001
From: Raphael Roberts <raphael.roberts48@gmail.com>
Date: Mon, 21 Oct 2019 12:08:29 -0500
Subject: [PATCH] Should fix diagonal_then_reversed

---
 prim_wallpaper/pickers/diagonal.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: