diff --git a/tests/integrations_app/apps/custom_work_dependencies/app.py b/tests/integrations_app/apps/custom_work_dependencies/app.py
index d09d3048b3..10f9992b9a 100644
--- a/tests/integrations_app/apps/custom_work_dependencies/app.py
+++ b/tests/integrations_app/apps/custom_work_dependencies/app.py
@@ -24,8 +24,7 @@ class WorkWithCustomBaseImage(LightningWork):
     def __init__(self, cloud_compute: CloudCompute = CloudCompute(), **kwargs):
         # this image has been created from ghcr.io/gridai/base-images:v1.8-cpu
         # by just adding an empty file at /content/.e2e_test
-        image_tag = "v1.15"
-        # image_tag = os.getenv("LIGHTNING_E2E_TEST_IMAGE_VERSION", "v1.15")
+        image_tag = os.getenv("LIGHTNING_E2E_TEST_IMAGE_VERSION", "v1.15")
         custom_image = f"ghcr.io/gridai/image-for-testing-custom-images-in-e2e:{image_tag}"
         build_config = BuildConfig(image=custom_image)
         super().__init__(parallel=True, **kwargs, cloud_compute=cloud_compute, cloud_build_config=build_config)