From 2322e5b6395c2c17088d79657ab33302c2e5e508 Mon Sep 17 00:00:00 2001
From: Chris Tate <chris@ctate.dev>
Date: Mon, 12 Jan 2026 01:19:14 -0600
Subject: [PATCH] address feedback

---
 cli/src/commands.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/cli/src/commands.rs b/cli/src/commands.rs
index 8b8ff58..405f753 100644
--- a/cli/src/commands.rs
+++ b/cli/src/commands.rs
@@ -223,7 +223,10 @@ pub fn parse_command(args: &[String], flags: &Flags) -> Result<Value, ParseError
             
             // Check for --load flag: wait --load networkidle
             if let Some(idx) = rest.iter().position(|&s| s == "--load" || s == "-l") {
-                let state = rest.get(idx + 1).unwrap_or(&"load");
+                let state = rest.get(idx + 1).ok_or_else(|| ParseError::MissingArguments {
+                    context: "wait --load".to_string(),
+                    usage: "wait --load <state>",
+                })?;
                 return Ok(json!({ "id": id, "action": "waitforloadstate", "state": state }));
             }
             
@@ -1155,6 +1158,13 @@ mod tests {
         assert_eq!(cmd["state"], "networkidle");
     }
 
+    #[test]
+    fn test_wait_load_missing_state() {
+        let result = parse_command(&args("wait --load"), &default_flags());
+        assert!(result.is_err());
+        assert!(matches!(result.unwrap_err(), ParseError::MissingArguments { .. }));
+    }
+
     #[test]
     fn test_wait_fn() {
         let cmd = parse_command(&args("wait --fn window.ready"), &default_flags()).unwrap();