diff --git a/src/main.rs b/src/main.rs
index 4d0c721..2ab289a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,10 +1,10 @@
-use advent_of_code::prob1;
+use advent_of_code::{prob1, prob2};
 
 fn main() {
     let args: Vec<String> = std::env::args().collect();
     let file_path = args
         .get(1)
         .expect("Please provide an argument on command-line.");
-    let max = prob1::solve_part2(file_path);
+    let max = prob2::solve_part1(file_path);
     println!("{max}");
 }