diff --git a/src/main.rs b/src/main.rs
index 0291b62..5e106ff 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,10 +1,10 @@
-use _1_calorie_counting::prob1::solve;
+use _1_calorie_counting::prob1;
 
 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 = solve(file_path);
+    let max = prob1::solve(file_path);
     println!("{max}");
 }