1 files changed, 4 insertions, 6 deletions
diff --git a/src/db/event_colors.rs b/src/db/event_colors.rs
index 62b6146..bf7f541 100644
--- a/src/db/event_colors.rs
+++ b/src/db/event_colors.rs
@@ -3,12 +3,10 @@ use rusqlite::Connection;
 
 pub fn get_default_color(conn: &Connection) -> Result<String> {
     let mut stmt = conn.prepare("SELECT * FROM event_colors LIMIT 1")?;
+    let mut iter = stmt.query_map([], |row| row.get(0))?;
 
-    let iter = stmt.query_map([], |row| row.get(0))?;
-
-    let mut res = vec![];
-    for color in iter {
-        res.push(color?)
+    match iter.next() {
+        Some(Ok(color)) => Ok(color),
+        _ => Ok("blue".to_string()),
     }
-    Ok(res.first().unwrap_or(&"blue".to_string()).clone())
 }