diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py
index 00b4be8..b2291f6 100644
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@@ -216,7 +216,7 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
         from pympler import muppy, summary
         all_objects = muppy.get_objects()
         all_lists = muppy.filter(all_objects, Type=list)
-        summary.print_(summary.summarize(all_objects))
+        summary.print_(summary.summarize(all_lists))
     
     epoch_end_time = time.time()
     per_epoch_time = epoch_end_time - epoch_start_time