diff --git a/cmd/api/main.go b/cmd/api/main.go
index 57f404aa4..0ae0ab094 100644
--- a/cmd/api/main.go
+++ b/cmd/api/main.go
@@ -29,7 +29,7 @@ var startCmd = &cobra.Command{
 func main() {
 	startCmd.Flags().IntP(constant.FlagPort, "p", 0, "Port to listen on")
 	startCmd.Flags().StringP(constant.FlagConf, "c", "", "Path to config file folder")
-	rootCmd := cmd.NewRootCmd()
+	rootCmd := cmd.NewRootCmd(nil)
 	rootCmd.Command.AddCommand(startCmd)
 	if err := startCmd.Execute(); err != nil {
 		fmt.Println(err)
diff --git a/pkg/common/cmd/root.go b/pkg/common/cmd/root.go
index f9af56a49..54bf28943 100644
--- a/pkg/common/cmd/root.go
+++ b/pkg/common/cmd/root.go
@@ -10,7 +10,7 @@ type RootCmd struct {
 	Command cobra.Command
 }
 
-func NewRootCmd() RootCmd {
+func NewRootCmd(f func() error) RootCmd {
 	c := cobra.Command{
 		Use:   "start",
 		Short: "Start the server",