diff --git a/tools/cmd/main.go b/tools/cmd/main.go
index be907c1b2..f936fdb90 100644
--- a/tools/cmd/main.go
+++ b/tools/cmd/main.go
@@ -11,7 +11,7 @@ import (
 	"kitty/tools/cmd/tool"
 )
 
-func main() {
+func KittenMain(args ...string) {
 	krm := os.Getenv("KITTY_KITTEN_RUN_MODULE")
 	os.Unsetenv("KITTY_KITTEN_RUN_MODULE")
 	switch krm {
@@ -31,5 +31,9 @@ func main() {
 	tool.KittyToolEntryPoints(root)
 	completion.EntryPoint(root)
 
-	root.Exec()
+	root.Exec(args...)
+}
+
+func main() {
+	KittenMain()
 }