diff --git a/src/main.rs b/src/main.rs
index 2338bde..58bbfb0 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -96,17 +96,20 @@ async fn createuser(
 }
 
 #[get("/account")]
-async fn account(mut db: Connection<Db>, cookies: &CookieJar<'_>) -> String {
+async fn account(mut db: Connection<Db>, cookies: &CookieJar<'_>) -> status::Custom<String> {
     let token = cookies.get_private("token");
     match token {
         Some(t) => match User::get_by_token(&mut db, t).await {
-            Some(user) => format!(
-                "Username: {}\nAdmin: {}\nMake Posts: {}\nComment: {}",
-                user.username, user.admin, user.make_posts, user.comment
+            Some(user) => status::Custom(
+                Status::Ok,
+                format!(
+                    "Username: {}\nAdmin: {}\nMake Posts: {}\nComment: {}",
+                    user.username, user.admin, user.make_posts, user.comment
+                ),
             ),
-            None => "User doesn't exist.".to_string(),
+            None => status::Custom(Status::NotFound, "User doesn't exist.".to_string()),
         },
-        None => "Not logged in".to_string(),
+        None => status::Custom(Status::Unauthorized, "Not logged in".to_string()),
     }
 }