diff --git a/main.go b/main.go
index 45f48f6a418f61d8c2f7ecfaf99210d310a3d659..f8a6e5a9b72dcad39ff0174eb12728039a6caf5c 100644
--- a/main.go
+++ b/main.go
@@ -56,13 +56,14 @@ var version = ""
 func main() {
 	configPath := "./config.toml"
 	for i, arg := range os.Args {
-		if arg == "-v" || arg == "--version" {
+		switch arg {
+		case "-v", "--version":
 			if version == "" {
 				version = "unknown, build with `just build` or copy/paste the build command from ./justfile"
 			}
 			fmt.Println("lunatask-mcp-server:", version)
 			os.Exit(0)
-		} else if arg == "-c" || arg == "--config" {
+		case "-c", "--config":
 			if i+1 < len(os.Args) {
 				configPath = os.Args[i+1]
 			}