diff --git a/core/kg/cmd/commands/root.go b/core/kg/cmd/commands/root.go
index 68831c3..812cdf1 100644
--- a/core/kg/cmd/commands/root.go
+++ b/core/kg/cmd/commands/root.go
@@ -16,7 +16,7 @@
 }
 
 // rootCmd is a command to run the server.
-var rootCmd = &cobra.Command{
+var rootCmd = &Command{
 	Use:   "server",
 	Short: "An example of the basic server",
 	RunE:  serverCmdF,