diff --git a/source/cli/main.cpp b/source/cli/main.cpp
index 85a31f3..3b88380 100644
--- a/source/cli/main.cpp
+++ b/source/cli/main.cpp
@@ -163,7 +163,7 @@ int main(int argc, char** argv)
 }
 
 bool parse_param(int argc, char** argv, std::unordered_map<std::string, std::string>& options,
-                 std::unordered_map<std::string, bool> flags)
+                 std::unordered_map<std::string, bool>& flags)
 {
     std::unordered_set<std::string> allowed_param = { "control", "device", "tasks", "init" };
     std::unordered_set<std::string> flags_key = { "init" };
diff --git a/source/cli/main.h b/source/cli/main.h
index ddc101a..492555c 100644
--- a/source/cli/main.h
+++ b/source/cli/main.h
@@ -20,7 +20,7 @@
 #include "tasks.h"
 
 bool parse_param(int argc, char** argv, std::unordered_map<std::string, std::string>& options,
-                 std::unordered_map<std::string, bool> flags);
+                 std::unordered_map<std::string, bool>& flags);
 
 void print_help();