diff --git a/2024/4/puzzle1.py b/2024/4/puzzle1.py
index 434b5ae..58689bc 100644
--- a/2024/4/puzzle1.py
+++ b/2024/4/puzzle1.py
@@ -63,11 +63,11 @@ if __name__ == "__main__":
 			result += scan_line(line[::-1])
 
 		diagonal_list = rotate(full_string)
-		for line in full_string:
+		for line in diagonal_list:
 			result += scan_line(line)
 			result += scan_line(line[::-1])
 		# TODO: Transposion after rotating is ill-defined.
-		for line in transpose(rotate(full_string)):
+		for line in rotate(transpose(full_string)):
 			result += scan_line(line)
 			result += scan_line(line[::-1])