diff --git a/main.go b/main.go
index 550936c..2766d99 100644
--- a/main.go
+++ b/main.go
@@ -158,8 +158,9 @@ func main() {
 	log.Infof("Redis connection established (%s)", ping_result)
 
 	command := ""
-	if len(os.Args) > 1 {
-		command = os.Args[1]
+	args := flag.Args()
+	if len(args) > 1 {
+		command = args[1]
 	}
 
 	switch command {