diff --git a/python/python/lance/dataset.py b/python/python/lance/dataset.py
index 9b5f4f0857..7344541cec 100644
--- a/python/python/lance/dataset.py
+++ b/python/python/lance/dataset.py
@@ -1526,7 +1526,7 @@ def create_index(
             kwargs["shuffle_partition_concurrency"] = shuffle_partition_concurrency
 
         self._ds.create_index(column, index_type, name, replace, kwargs)
-        return LanceDataset(self.uri, index_cache_size=index_cache_size)
+        return self
 
     def session(self) -> Session:
         """