Use sol_path

1 files changed, 4 insertions, 8 deletions

diff --git a/src/app_state.rs b/src/app_state.rs
index d7de1fd..1000047 100644
--- a/src/app_state.rs
+++ b/src/app_state.rs
@@ -321,14 +321,10 @@ impl AppState {
                 .write_solution_to_disk(self.current_exercise_ind, current_exercise.name)
                 .map(Some)
         } else {
-            let solution_path = if let Some(dir) = current_exercise.dir {
-                format!("solutions/{dir}/{}.rs", current_exercise.name)
-            } else {
-                format!("solutions/{}.rs", current_exercise.name)
-            };
-
-            if Path::new(&solution_path).exists() {
-                return Ok(Some(solution_path));
+            let sol_path = current_exercise.sol_path();
+
+            if Path::new(&sol_path).exists() {
+                return Ok(Some(sol_path));
             }
 
             Ok(None)