diff --git a/tests/openapi/test_openapi.py b/tests/openapi/test_openapi.py
index a70227e35..015f60873 100644
--- a/tests/openapi/test_openapi.py
+++ b/tests/openapi/test_openapi.py
@@ -292,7 +292,7 @@ def setUp(self):
     def test_must_iterate_on_paths(self):
 
         expected = {"/foo", "/bar", "/baz"}
-        actual = set([path for path in self.editor.iter_on_path()])
+        actual = set(list(self.editor.iter_on_path()))
 
         self.assertEqual(expected, actual)
 
diff --git a/tests/swagger/test_swagger.py b/tests/swagger/test_swagger.py
index 615fabec3..cf6147726 100644
--- a/tests/swagger/test_swagger.py
+++ b/tests/swagger/test_swagger.py
@@ -302,7 +302,7 @@ def setUp(self):
     def test_must_iterate_on_paths(self):
 
         expected = {"/foo", "/bar", "/baz"}
-        actual = set([path for path in self.editor.iter_on_path()])
+        actual = set(list(self.editor.iter_on_path()))
 
         self.assertEqual(expected, actual)