diff --git a/bin/omarchy-state b/bin/omarchy-state
index eec79584..d55943b6 100755
--- a/bin/omarchy-state
+++ b/bin/omarchy-state
@@ -6,24 +6,17 @@ mkdir -p "$STATE_DIR"
 COMMAND="$1"
 STATE_NAME="$2"
 
-case "$COMMAND" in
-set)
-  if [[ -z "$STATE_NAME" ]]; then
-    echo "Usage: omarchy-state set <state-name>" >&2
-    exit 1
-  fi
-  touch "$STATE_DIR/$STATE_NAME"
-  ;;
-clear)
-  if [[ -z "$STATE_NAME" ]]; then
-    echo "Usage: omarchy-state clear <state-name-or-pattern>" >&2
-    exit 1
-  fi
-  # Use find with -maxdepth 1 to match files in STATE_DIR
-  find "$STATE_DIR" -maxdepth 1 -type f -name "$STATE_NAME" -delete
-  ;;
-*)
-  echo "Usage: omarchy-state <set|clear> <state-name-or-pattern>" >&2
+if [[ -z "$COMMAND" ]]; then
+  echo "Usage: omarchy-state <set|clear> <state-name-or-pattern>"
   exit 1
-  ;;
+fi
+
+if [[ -z "$STATE_NAME" ]]; then
+  echo "Usage: omarchy-state $COMMAND <state-name>"
+  exit 1
+fi
+
+case "$COMMAND" in
+set) touch "$STATE_DIR/$STATE_NAME" ;;
+clear) find "$STATE_DIR" -maxdepth 1 -type f -name "$STATE_NAME" -delete ;;
 esac