diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py
index 905c8fa..0351fb0 100644
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@@ -203,7 +203,7 @@ def predict_keras(generator: callable,
             # do observation stuff
             predictions = np.asarray(_get_observations(detections))
         else:
-            predictions = ssd_model.predict_on_batch(x)
+            predictions = np.asarray(ssd_model.predict_on_batch(x))
             print(predictions[:, :, [-2, -1]])
         
         decoded_predictions_batch = ssd_output_decoder.decode_detections_fast(