diff --git a/cmd/hf/main.go b/cmd/hf/main.go
index 98a51d2..5175af5 100644
--- a/cmd/hf/main.go
+++ b/cmd/hf/main.go
@@ -68,6 +68,11 @@ func main() {
 		default:
 			output.Errorf("unknown agent subcommand: %s", args[1])
 		}
+	case "assign-schedule-type":
+		// Leaf command (no subcommands) — args are <agent-id> <type-name>.
+		// Must dispatch at top level because handleGroup treats args[0] as
+		// a subcommand name and would error "unknown ... subcommand: <agent-id>".
+		handleAssignScheduleType(args[1:])
 	default:
 		if group, ok := findGroup(args[0]); ok {
 			handleGroup(group, args[1:])