diff --git a/test/test_database.py b/test/test_database.py
index 4bef9e59857dcd15f52a3f22c6b89fa35e3c6065..ff2bb6b7fc4bbb2b01f6febb1c0dbd0b6090fb83 100644
--- a/test/test_database.py
+++ b/test/test_database.py
@@ -119,7 +119,8 @@ class DatabaseTest(unittest.TestCase):
             line = f.readline().split(", ")
             categories_file.append(line[0])
             # Skip all words:
-            [f.readline() for _ in range(int(line[1].rstrip()))]
+            for _ in range(int(line[1].rstrip())):
+                f.readline()
 
         f.close()