diff --git a/userdb/db.py b/userdb/db.py
index 413e159..37353e2 100644
--- a/userdb/db.py
+++ b/userdb/db.py
@@ -17,7 +17,7 @@ app = Flask(__name__)
 
 @app.route("/track", methods = ['GET'])
 def track():
-    if request.method == 'GET' and request.args.get('hash'):
+    if request.method == 'GET' and request.args.get('hash') and request.args.get('cat'):
         
         hash = str(request.args.get('hash'))
         cat = str(request.args.get('cat'))
@@ -30,7 +30,7 @@ def track():
         else:
             db = {}
         if hash in db.keys():
-            visits = db[hash]['visits']
+            visits = db[hash][cat]
         else:
             visits = list()