diff --git a/main.py b/main.py
index 7645ee4..e62672f 100644
--- a/main.py
+++ b/main.py
@@ -84,9 +84,9 @@ def get_categories():
         instead of returning lists.
 
     :return: The list of video categories
-    :rtype: list
+    :rtype: types.GeneratorType
     """
-    return VIDEOS.keys()
+    return VIDEOS.iterkeys()
 
 
 def get_videos(category):