diff --git a/zsh_completions_test.go b/zsh_completions_test.go
index b1321c9..22a66d6 100644
--- a/zsh_completions_test.go
+++ b/zsh_completions_test.go
@@ -134,6 +134,18 @@ func TestGenZshCompletion(t *testing.T) {
 			},
 			skip: "--version and --help are currently not generated when not running on root command",
 		},
+		{
+			name: "zsh generation should run on root commannd",
+			root: func() *Command {
+				r := genTestCommand("root", false)
+				s := genTestCommand("sub1", true)
+				r.AddCommand(s)
+				return s
+			}(),
+			expectedExpressions: []string{
+				"function _root {",
+			},
+		},
 	}
 
 	for _, tc := range tcs {