diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py
index d413370..6aa44f9 100644
--- a/src/twomartens/masterthesis/data.py
+++ b/src/twomartens/masterthesis/data.py
@@ -22,7 +22,7 @@ Functions:
     load_coco_val(...): loads the COCO validation data into a Tensorflow data set
     load_scenenet(...): loads the SceneNet RGB-D data into a Tensorflow data set
 """
-from typing import Callable, List, Mapping, Tuple, Any, Dict
+from typing import Callable, List, Mapping, Tuple
 from typing import Sequence
 
 import numpy as np