Update cmd/cli/desktop/desktop.go

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: ericcurtin

cmd/cli/desktop/desktop.go

@@ -524,16 +524,17 @@ func (c *Client) Remove(modelArgs []string, force bool) (string, error) {
 	modelRemoved := ""
 	for _, model := range modelArgs {
 		// First, check if this looks like a model ID (no slash) - before normalizing
+		isExpandedAsID := false
 		if !strings.Contains(model, "/") {
 			if expanded, err := c.fullModelID(model); err == nil {
 				// It's a model ID, use the expanded form
 				model = expanded
-			} else {
-				// Not a model ID, normalize normally
-				model = dmrm.NormalizeModelName(model)
+				isExpandedAsID = true
 			}
-		} else {
-			// Has slash, normalize normally
+		}
+
+		if !isExpandedAsID {
+			// Not a valid model ID or has a slash, so normalize as a model name
 			model = dmrm.NormalizeModelName(model)
 		}