diff --git a/tests/test_image_robustness.py b/tests/test_image_robustness.py
index 5b9843d26..85142937e 100644
--- a/tests/test_image_robustness.py
+++ b/tests/test_image_robustness.py
@@ -2,9 +2,7 @@
 from PIL import Image
 import requests
 
-from langtest.transform.image import robustness as image_robustness
 from langtest.transform.robustness import BaseRobustness
-
 from langtest.utils.custom_types.sample import VisualQASample