diff --git a/tests/test_masked_inference_wsi_dataset.py b/tests/test_masked_inference_wsi_dataset.py
index 7dc322cefa..927ba9de5f 100644
--- a/tests/test_masked_inference_wsi_dataset.py
+++ b/tests/test_masked_inference_wsi_dataset.py
@@ -22,7 +22,7 @@
 from monai.utils import optional_import
 from tests.utils import skip_if_quick
 
-_, has_cim = optional_import("cucim")
+_, has_cim = optional_import("cucim", name="CuImage")
 _, has_osl = optional_import("openslide")
 
 FILE_URL = "https://drive.google.com/uc?id=1sGTKZlJBIz53pfqTxoTqiIQzIoEzHLAe"