diff --git a/src/cli/nomos_cli.py b/src/cli/nomos_cli.py
index d4adb9b..f13255c 100644
--- a/src/cli/nomos_cli.py
+++ b/src/cli/nomos_cli.py
@@ -16,7 +16,14 @@ logger = get_custom_logger(__name__)
 
 
 class NomosCli:
-    def __init__(self, command=""):
+    def __init__(self, **kwargs):
+        if "command" not in kwargs:
+            raise ValueError("The command parameter is required")
+
+        command = kwargs["command"]
+        if command not in nomos_cli:
+            raise ValueError("Unknown command provided")
+
         logger.debug(f"Cli is going to be initialized with this config {nomos_cli[command]}")
         self._command = command
         self._image_name = nomos_cli[command]["image"]