diff --git a/day-05/day-05.py b/day-05/day-05.py
index 31167e8..6b2c0c9 100644
--- a/day-05/day-05.py
+++ b/day-05/day-05.py
@@ -35,12 +35,8 @@ def part_2(input):
                 for y in range(min(y1,y2), max(y1,y2) + 1):
                     field[x][y] += 1
         elif abs(x1 - x2) == abs(y1 - y2):
-            dx = 1
-            if x1 > x2:
-                dx = -1
-            dy = 1
-            if y1 > y2:
-                dy = -1
+            dx = 1 if x1 < x2 else -1
+            dy = 1 if y1 < y2 else -1
             for n in range(abs(x1 - x2) + 1):
                 field[x1 + (dx * n)][y1 + (dy * n)] += 1
     for x in range(1000):