diff --git a/pydelmod/dataui.py b/pydelmod/dataui.py
index 52e171f..762f883 100644
--- a/pydelmod/dataui.py
+++ b/pydelmod/dataui.py
@@ -246,12 +246,12 @@ def update_map_features(
                 )
             ]
         else:
-            dfs = dfs.loc[self.display_table.current_view.index]
-            merged_view = self.display_table.selected_dataframe.merge(
-                self.display_table.current_view.reset_index(drop=True).reset_index(),
-                how="inner",
-            )
-            current_selection = merged_view["index"].to_list()
+            dfs = dfs.iloc[self.display_table.current_view.index]
+        merged_view = self.display_table.selected_dataframe.merge(
+            self.display_table.current_view.reset_index(drop=True).reset_index(),
+            how="inner",
+        )
+        current_selection = merged_view["index"].to_list()
 
         try:
             if len(query) > 0: