diff --git a/lightly/transforms/torchvision_transforms.py b/lightly/transforms/torchvision_transforms.py
index 20e5d8f60..6e26f571b 100644
--- a/lightly/transforms/torchvision_transforms.py
+++ b/lightly/transforms/torchvision_transforms.py
@@ -5,7 +5,7 @@
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 #
-from typing import Callable, Generic, Protocol, TypeVar, Union
+from typing import Union
 
 import torch
 from PIL.Image import Image