diff --git a/src/lightning/data/datasets/index.py b/src/lightning/data/datasets/index.py
index 34ecaaf095cc4..d223b90d4da07 100644
--- a/src/lightning/data/datasets/index.py
+++ b/src/lightning/data/datasets/index.py
@@ -2,7 +2,7 @@
 import os
 from io import TextIOWrapper
 
-from lightning.app.utilities.network import LightningClient
+from lightning_cloud.rest_client import LightningClient
 
 
 def get_index(s3_connection_path: str, index_file_path: str) -> bool: