diff --git a/internal/cli/serverless/branch/create_test.go b/internal/cli/serverless/branch/create_test.go
index 8865885a..99abe9dd 100644
--- a/internal/cli/serverless/branch/create_test.go
+++ b/internal/cli/serverless/branch/create_test.go
@@ -91,17 +91,17 @@ func (suite *CreateBranchSuite) TestCreateBranchArgs() {
 	}{
 		{
 			name:         "create branch success",
-			args:         []string{"--cluster-id", clusterID, "--branch-name", branchName},
+			args:         []string{"--cluster-id", clusterID, "--display-name", branchName},
 			stdoutString: fmt.Sprintf("... Waiting for branch to be ready\nBranch %s is ready.", branchId),
 		},
 		{
 			name:         "create branch with shorthand flag",
-			args:         []string{"-c", clusterID, "--branch-name", branchName},
+			args:         []string{"-c", clusterID, "--display-name", branchName},
 			stdoutString: fmt.Sprintf("... Waiting for branch to be ready\nBranch %s is ready.", branchId),
 		},
 		{
 			name: "without required project id",
-			args: []string{"--branch-name", branchName},
+			args: []string{"--display-name", branchName},
 			err:  fmt.Errorf("required flag(s) \"cluster-id\" not set"),
 		},
 	}
diff --git a/internal/cli/serverless/create_test.go b/internal/cli/serverless/create_test.go
index 04df2e94..a770ae76 100644
--- a/internal/cli/serverless/create_test.go
+++ b/internal/cli/serverless/create_test.go
@@ -97,12 +97,12 @@ func (suite *CreateClusterSuite) TestCreateClusterArgs() {
 	}{
 		{
 			name:         "create cluster success",
-			args:         []string{"--project-id", projectID, "--cluster-name", clusterName, "--cloud-provider", cloudProvider, "--region", region},
+			args:         []string{"--project-id", projectID, "--display-name", clusterName, "--cloud-provider", cloudProvider, "--region", region},
 			stdoutString: "... Waiting for cluster to be ready\nCluster 12345 is ready.",
 		},
 		{
 			name:         "create cluster with shorthand flag",
-			args:         []string{"-p", projectID, "--cluster-name", clusterName, "--cloud-provider", cloudProvider, "-r", region},
+			args:         []string{"-p", projectID, "--display-name", clusterName, "--cloud-provider", cloudProvider, "-r", region},
 			stdoutString: "... Waiting for cluster to be ready\nCluster 12345 is ready.",
 		},
 	}
@@ -168,7 +168,7 @@ func (suite *CreateClusterSuite) TestCreateClusterWithoutProject() {
 	}{
 		{
 			name:         "without project id",
-			args:         []string{"--cluster-name", clusterName, "--cloud-provider", cloudProvider, "-r", region},
+			args:         []string{"--display-name", clusterName, "--cloud-provider", cloudProvider, "-r", region},
 			stdoutString: "... Waiting for cluster to be ready\nCluster 12345 is ready.",
 		},
 	}
diff --git a/internal/cli/serverless/update_test.go b/internal/cli/serverless/update_test.go
index e888d23e..262d6d12 100644
--- a/internal/cli/serverless/update_test.go
+++ b/internal/cli/serverless/update_test.go
@@ -80,7 +80,7 @@ func (suite *UpdateClusterSuite) TestUpdateClusterArgs() {
 	}{
 		{
 			name:         "update displayName success",
-			args:         []string{"--cluster-id", clusterID, "--cluster-name", displayName},
+			args:         []string{"--cluster-id", clusterID, "--display-name", displayName},
 			stdoutString: fmt.Sprintf("cluster %s updated\n", clusterID),
 		},
 	}