diff --git a/cmd/cmd.go b/cmd/cmd.go
index a26ec5a0..10e6b6a9 100644
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -102,16 +102,14 @@ func RunHandler(cmd *cobra.Command, args []string) error {
 	name := args[0]
 	// check if the model exists on the server
 	_, err = client.Show(context.Background(), &api.ShowRequest{Name: name})
-	if err != nil {
-		var statusError api.StatusError
-		switch {
-		case errors.As(err, &statusError) && statusError.StatusCode == http.StatusNotFound:
-			if err := PullHandler(cmd, args); err != nil {
-				return err
-			}
-		case err != nil:
+	var statusError api.StatusError
+	switch {
+	case errors.As(err, &statusError) && statusError.StatusCode == http.StatusNotFound:
+		if err := PullHandler(cmd, args); err != nil {
 			return err
 		}
+	case err != nil:
+		return err
 	}
 
 	return RunGenerate(cmd, args)