diff --git a/cmd/actions/generate.go b/cmd/actions/generate.go
index 38739a8..c4b12ad 100644
--- a/cmd/actions/generate.go
+++ b/cmd/actions/generate.go
@@ -9,26 +9,21 @@ import (
 
 var (
 	generateCmd = &cobra.Command{
-		Use:     "generate [UNIT...]",
+		Use:     "generate",
 		Short:   "generate and view the squadron chart",
 		Example: "  squadron generate fronted backend",
 		Args:    cobra.MinimumNArgs(0),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			return generate(log, args, cwd, flagFiles)
+			return generate(log, cwd, flagFiles)
 		},
 	}
 )
 
-func generate(l *logrus.Entry, args []string, cwd string, files []string) error {
+func generate(l *logrus.Entry, cwd string, files []string) error {
 	sq, err := squadron.New(l, cwd, "", files)
 	if err != nil {
 		return err
 	}
 
-	units, err := parseUnitArgs(args, sq.GetUnits())
-	if err != nil {
-		return err
-	}
-
-	return sq.Generate(units)
+	return sq.Generate(sq.GetUnits())
 }
diff --git a/cmd/actions/up.go b/cmd/actions/up.go
index a0505ba..91303a4 100644
--- a/cmd/actions/up.go
+++ b/cmd/actions/up.go
@@ -51,7 +51,7 @@ func up(l *logrus.Entry, args []string, cwd, namespace string, build, push bool,
 		}
 	}
 
-	if err := sq.Generate(units); err != nil {
+	if err := sq.Generate(sq.GetUnits()); err != nil {
 		return err
 	} else if err := sq.Up(helmArgs); err != nil {
 		return err