diff --git a/steam_chat_viewer/__init__.py b/steam_chat_viewer/__init__.py
index 76e2242..c1b7be7 100644
--- a/steam_chat_viewer/__init__.py
+++ b/steam_chat_viewer/__init__.py
@@ -11,8 +11,8 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return str(row for row in query(
+    return str(query(
         '''
             SELECT * FROM messages
         '''
-    ).rows)
+    ).rowcount)