diff --git a/app/Commands/DaemonLogsCommand.php b/app/Commands/DaemonLogsCommand.php
index e538767..219fe1c 100644
--- a/app/Commands/DaemonLogsCommand.php
+++ b/app/Commands/DaemonLogsCommand.php
@@ -28,7 +28,7 @@ class DaemonLogsCommand extends Command
      */
     public function handle()
     {
-        $daemonId = $this->askForDaemon('Which daemon would you like to retrieve the logs from');
+        $daemonId = $this->argument('daemon') ?? $this->askForDaemon('Which daemon would you like to retrieve the logs from');
 
         $daemon = $this->forge->daemon($this->currentServer()->id, $daemonId);
 
diff --git a/app/Commands/DaemonRestartCommand.php b/app/Commands/DaemonRestartCommand.php
index 5b2b6f1..661b823 100644
--- a/app/Commands/DaemonRestartCommand.php
+++ b/app/Commands/DaemonRestartCommand.php
@@ -27,7 +27,7 @@ public function handle()
     {
         $server = $this->currentServer();
 
-        $daemonId = $this->askForDaemon('Which daemon would you like to restart');
+        $daemonId = $this->argument('daemon') ?? $this->askForDaemon('Which daemon would you like to restart');
 
         $daemon = $this->forge->daemon($server->id, $daemonId);