diff --git a/tests/utilities/test_xla_device_utils.py b/tests/utilities/test_xla_device_utils.py
index 9bcb4f8dea..73b11b4826 100644
--- a/tests/utilities/test_xla_device_utils.py
+++ b/tests/utilities/test_xla_device_utils.py
@@ -16,7 +16,7 @@ from unittest.mock import patch
 
 import pytest
 
-import pytorch_lightning.utilities.xla_device_utils as xla_utils
+import pytorch_lightning.utilities.xla_device as xla_utils
 from pytorch_lightning.utilities import _TPU_AVAILABLE, _XLA_AVAILABLE
 from tests.helpers.utils import pl_multi_process_test
 
@@ -34,7 +34,7 @@ def test_tpu_device_presence():
     assert xla_utils.XLADeviceUtils.tpu_device_exists() is True
 
 
-@patch('pytorch_lightning.utilities.xla_device_utils.TPU_CHECK_TIMEOUT', 10)
+@patch('pytorch_lightning.utilities.xla_device.TPU_CHECK_TIMEOUT', 10)
 def test_result_returns_within_timeout_seconds():
     """Check that pl_multi_process returns within 10 seconds"""
     start = time.time()