diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py
index bcbb779..cb5e61b 100644
--- a/src/twomartens/masterthesis/main.py
+++ b/src/twomartens/masterthesis/main.py
@@ -173,6 +173,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
     from twomartens.masterthesis import ssd
     
     tf.enable_eager_execution()
+    tf.enable_v2_tensorshape()
     batch_size = 16
     image_size = 256
     use_dropout = False