From 3833a3e2ac71a6ffc44cc855cd1e147717b7de7e Mon Sep 17 00:00:00 2001
From: Roman Miroshnychenko <roman1972@gmail.com>
Date: Mon, 20 Nov 2017 16:37:03 +0200
Subject: [PATCH] Use iterkeys() to iterate dict keys

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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):