diff --git a/python/ray/dataframe/indexing.py b/python/ray/dataframe/indexing.py
index 5f99ce877323..2fa2e1e8d2c2 100644
--- a/python/ray/dataframe/indexing.py
+++ b/python/ray/dataframe/indexing.py
@@ -370,7 +370,11 @@ def _compute_enlarge_labels(self, locator, base_index):
         Returns:
              nan_labels: The labels needs to be added
         """
-        locator_as_index = pandas.Index(locator)
+        # base_index_type can be pd.Index or pd.DatetimeIndex
+        # depending on user input and pandas behavior
+        # See issue #2264
+        base_index_type = type(base_index)
+        locator_as_index = base_index_type(locator)
 
         nan_labels = locator_as_index.difference(base_index)
         common_labels = locator_as_index.intersection(base_index)