diff --git a/src/day1.rs b/src/day1.rs
index 6f9b66f..e25f57b 100644
--- a/src/day1.rs
+++ b/src/day1.rs
@@ -11,12 +11,29 @@ mod part1 {
     #[test]
     fn run_sample() {
         let input = include_str!("../inputs/1_test.txt");
-        assert_eq!(part_1_solution(input), 0);
+        assert_eq!(part_1_solution(input), 7);
     }
 
     #[test]
     fn run() {
         let input = include_str!("../inputs/1.txt");
-        assert_eq!(part_1_solution(input), 0);
+        assert_eq!(part_1_solution(input), 1139);
+    }
+}
+
+#[cfg(test)]
+mod part2 {
+    use super::*;
+
+    #[test]
+    fn run_sample() {
+        let input = include_str!("../inputs/1_test.txt");
+        assert_eq!(part_2_solution(input), 5);
+    }
+
+    #[test]
+    fn run() {
+        let input = include_str!("../inputs/1.txt");
+        assert_eq!(part_2_solution(input), 1103);
     }
 }