1 files changed, 5 insertions, 2 deletions
diff --git a/04.c b/04.c
index 7077b65..499461b 100644
--- a/04.c
+++ b/04.c
@@ -5,6 +5,9 @@
 #include "input.h"
 
 #define INPUT "input/04.txt"
+#define EXPECTED1 51776L
+#define EXPECTED2 16830L
+
 #define BOARD_COUNT 100
 #define BOARD_SIZE 5
 #define MAX_DRAW 500
@@ -104,7 +107,7 @@ void part1(struct input_str* input) {
 			if(board_mark(&(board_list[j]), draw_list[i])) {
 				// winning board found, print score
 				long score = board_unmarked_sum(&(board_list[j]));
-				printf("%ld\n", draw_list[i] * score);
+				CHECK(draw_list[i] * score, EXPECTED1)
 				return;
 			}
 		}
@@ -134,7 +137,7 @@ void part2(struct input_str* input) {
 		}
 	}
 	// print last winning
-	printf("%ld\n", last_score);	
+	CHECK(last_score, EXPECTED2)
 }
 
 int main() {