diff --git a/src/config/session.rs b/src/config/session.rs
index 42f19c8..a03ba55 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -692,7 +692,7 @@ mod tests {
         let app_config = Arc::new(AppConfig::default());
         let app_state = Arc::new(AppState {
             config: app_config.clone(),
-            vault: Arc::new(crate::vault::Vault::default()),
+            vault: Arc::new(Vault::default()),
             mcp_factory: Arc::new(mcp_factory::McpFactory::default()),
             rag_cache: Arc::new(rag_cache::RagCache::default()),
             mcp_config: None,
diff --git a/src/utils/input.rs b/src/utils/input.rs
index ca59db2..e267bbd 100644
--- a/src/utils/input.rs
+++ b/src/utils/input.rs
@@ -21,8 +21,8 @@ pub fn read_single_key(valid_chars: &[char], default: char, prompt: &str) -> Res
                     break Err(anyhow::anyhow!("Interrupted"));
                 }
                 KeyCode::Char(c) if valid_chars.contains(&c) => {
-                    break Ok(c);
                     // Invalid character, continue loop
+                    break Ok(c);
                 }
                 KeyCode::Enter => {
                     break Ok(default);