diff --git a/project/src/config.rs b/project/src/config.rs
index b6714aa..a3a9fe7 100644
--- a/project/src/config.rs
+++ b/project/src/config.rs
@@ -13,7 +13,7 @@ pub fn get_operation_mode() -> Option<OperationMode> {
     let args: Vec<String> = env::args().collect();
 
     // Getting operation mode
-    match &args[1] {
+    match args[1].as_str() {
         "training" => Some(OperationMode::Training),
         "infrence" => Some(OperationMode::Infrence),
         _ => None,