diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py
index 244b339..7b60571 100644
--- a/src/twomartens/masterthesis/data.py
+++ b/src/twomartens/masterthesis/data.py
@@ -347,7 +347,7 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \
             
             return image_resized, _labels
     
-        processed = tf.map_fn(_get_images, (_images, labels), dtype=tf.float32)
+        processed = tf.map_fn(_get_images, (_images, labels), dtype=[tf.float32, tf.float32])
         processed_images = tf.reshape(processed[0], [-1, resized_shape[0], resized_shape[1], 3])
     
         return processed_images, processed[1]