diff --git a/cleanlab_studio/internal/api/api.py b/cleanlab_studio/internal/api/api.py
index 36f1bd02..383c038c 100644
--- a/cleanlab_studio/internal/api/api.py
+++ b/cleanlab_studio/internal/api/api.py
@@ -290,7 +290,7 @@ def download_cleanlab_columns(
     cleanset_json_io = io.StringIO(cleanset_json)
     cleanset_pd: pd.DataFrame = pd.read_json(cleanset_json_io, orient="table")
     cleanset_pd.rename(columns={"id": id_col}, inplace=True)
-    cleanset_pd.sort_values(by=id_col, inplace=True)
+    cleanset_pd.sort_values(by=id_col, inplace=True, ignore_index=True)
     return cleanset_pd
 
 
diff --git a/cleanlab_studio/studio/studio.py b/cleanlab_studio/studio/studio.py
index 834a4ba1..4e7122e2 100644
--- a/cleanlab_studio/studio/studio.py
+++ b/cleanlab_studio/studio/studio.py
@@ -138,11 +138,6 @@ def download_cleanlab_columns(
             include_project_details=include_project_details,
             to_spark=to_spark,
         )
-        if "cleanlab_row_ID" in rows_df.columns:
-            if to_spark:
-                rows_df.sort("cleanlab_row_ID")
-            else:
-                rows_df.sort_values(by="cleanlab_row_ID")
         return rows_df
 
     def apply_corrections(self, cleanset_id: str, dataset: Any, keep_excluded: bool = False) -> Any:
diff --git a/cleanlab_studio/version.py b/cleanlab_studio/version.py
index 50be1b5e..1ae40d9e 100644
--- a/cleanlab_studio/version.py
+++ b/cleanlab_studio/version.py
@@ -1,4 +1,4 @@
-__version__ = "2.0.0"
+__version__ = "2.0.1"
 
 SCHEMA_VERSION = "0.2.0"
 MIN_SCHEMA_VERSION = "0.1.0"