diff --git a/src/main.rs b/src/main.rs
index f1bc202..6e9a265 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,13 +1,13 @@
 use std::error::Error;
 
-mod websocket;
 mod computer;
 mod notifications;
+mod websocket;
 
 fn main() -> Result<(), Box<dyn Error>> {
     dotenvy::dotenv()?;
 
-    let mut socket = websocket::connect().unwrap();
+    let mut socket = websocket::connect().expect("Could not connect to WebSocket");
     computer::start_sending(&mut socket);
 
     Ok(())