diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py
index ea67826..cfc2f66 100644
--- a/src/twomartens/masterthesis/data.py
+++ b/src/twomartens/masterthesis/data.py
@@ -275,9 +275,11 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]],
             final_labels.append(labels)
         
     length_dataset = len(final_image_paths)
+    labels_np = np.array(final_labels)
+    print(labels_np.shape)
     
     path_dataset = tf.data.Dataset.from_tensor_slices(final_image_paths)
-    label_dataset = tf.data.Dataset.from_tensor_slices(np.array(final_labels))
+    label_dataset = tf.data.Dataset.from_tensor_slices(final_labels)
     dataset = tf.data.Dataset.zip((path_dataset, label_dataset))
     dataset = dataset.apply(tf.data.experimental.shuffle_and_repeat(buffer_size=length_dataset, count=num_epochs))
     dataset = dataset.batch(batch_size=batch_size)