diff --git a/test/unit_tests/test_predictor.py b/test/unit_tests/test_predictor.py
index 707c9a2..d35eb35 100644
--- a/test/unit_tests/test_predictor.py
+++ b/test/unit_tests/test_predictor.py
@@ -1,12 +1,11 @@
 def test_predict_pdf_works(predictor, test_pdf):
     # FIXME strange outputs, with '\n's in the dict
     predictions, metadata = predictor.predict_pdf(test_pdf)
-    predictions = list(predictions)
-    predictions = dict(**predictions[0])
-    assert predictions == {'class': 'formula',
-                           'probabilities': {'formula': 1.0, 'other': 1.576210689757148e-17,
-                                             'signature': 5.364939675629517e-24,
-                                             'logo': 8.815339177836606e-25}}
+    predictions = [p for p in predictions]
+    assert predictions[0] == {'class': 'formula',
+                              'probabilities': {'formula': 1.0, 'other': 1.576210689757148e-17,
+                                                'signature': 5.364939675629517e-24,
+                                                'logo': 8.815339177836606e-25}}
     metadata = list(metadata)
     metadata = dict(**metadata[0])
     metadata.pop("document_filename")  # temp filename cannot be tested