diff --git a/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_flux_2_lora_angles.rs b/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_flux_2_lora_angles.rs
index 599645574f..5328bedb60 100644
--- a/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_flux_2_lora_angles.rs
+++ b/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_flux_2_lora_angles.rs
@@ -54,13 +54,13 @@ fn resolve_single_image_input(
       }))
     }
     Some(ImageListRef::MediaFileTokens(tokens)) => {
-      if tokens.len() != 1 {
-        return Err(ArtcraftRouterError::Client(ClientError::ModelDoesNotSupportOption {
-          field: "image_inputs",
-          value: format!("Angle models require exactly one input image, got {}", tokens.len()),
-        }));
+      if let Some(token) = tokens.first() {
+        return Ok(token.clone());
       }
-      Ok(tokens.into_iter().next().unwrap())
+      Err(ArtcraftRouterError::Client(ClientError::ModelDoesNotSupportOption {
+        field: "image_inputs",
+        value: format!("Angle models require exactly one input image, got {}", tokens.len()),
+      }))
     }
     Some(ImageListRef::Urls(_)) => {
       Err(ArtcraftRouterError::Client(ClientError::ArtcraftOnlySupportsMediaTokens))
diff --git a/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_qwen_edit_2511_angles.rs b/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_qwen_edit_2511_angles.rs
index d439652f3c..b20cc80e62 100644
--- a/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_qwen_edit_2511_angles.rs
+++ b/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_qwen_edit_2511_angles.rs
@@ -56,13 +56,13 @@ fn resolve_single_image_input(
       }))
     }
     Some(ImageListRef::MediaFileTokens(tokens)) => {
-      if tokens.len() != 1 {
-        return Err(ArtcraftRouterError::Client(ClientError::ModelDoesNotSupportOption {
-          field: "image_inputs",
-          value: format!("Angle models require exactly one input image, got {}", tokens.len()),
-        }));
+      if let Some(token) = tokens.first() {
+        return Ok(token.clone());
       }
-      Ok(tokens.into_iter().next().unwrap())
+      Err(ArtcraftRouterError::Client(ClientError::ModelDoesNotSupportOption {
+        field: "image_inputs",
+        value: format!("Angle models require exactly one input image, got {}", tokens.len()),
+      }))
     }
     Some(ImageListRef::Urls(_)) => {
       Err(ArtcraftRouterError::Client(ClientError::ArtcraftOnlySupportsMediaTokens))