diff --git a/tests/test_image_utils.py b/tests/test_image_utils.py
index e6da9ceed2abcd7442a144324b298ef232340982..18c7fa24714951b7c842149764aee2b68dde0c7e 100644
--- a/tests/test_image_utils.py
+++ b/tests/test_image_utils.py
@@ -6,9 +6,8 @@
 import numpy
 import PIL.Image
 
-from ptbench.data.image_utils import (
+from ptbench.data.image_utils import (  # SingleAutoLevel16to8,
     RemoveBlackBorders,
-    SingleAutoLevel16to8,
     load_pil,
 )