diff --git a/packages/kbn-unified-data-table/src/components/default_cell_actions.tsx b/packages/kbn-unified-data-table/src/components/default_cell_actions.tsx
index 51730aef443497a..f2260a7d910deea 100644
--- a/packages/kbn-unified-data-table/src/components/default_cell_actions.tsx
+++ b/packages/kbn-unified-data-table/src/components/default_cell_actions.tsx
@@ -50,7 +50,7 @@ export const FilterInBtn = ({
 }) => {
   const context = useContext(UnifiedDataTableContext);
   const filteringDisabled =
-    isPlainRecord && Array.isArray(context.rows[rowIndex]?.flattened[columnId]);
+    isPlainRecord && Array.isArray(context.getRowByIndex(rowIndex)?.flattened[columnId]);
   const buttonTitle = i18n.translate('unifiedDataTable.grid.filterForAria', {
     defaultMessage: 'Filter for this {value}',
     values: { value: columnId },
@@ -85,7 +85,7 @@ export const FilterOutBtn = ({
 }) => {
   const context = useContext(UnifiedDataTableContext);
   const filteringDisabled =
-    isPlainRecord && Array.isArray(context.rows[rowIndex]?.flattened[columnId]);
+    isPlainRecord && Array.isArray(context.getRowByIndex(rowIndex)?.flattened[columnId]);
   const buttonTitle = i18n.translate('unifiedDataTable.grid.filterOutAria', {
     defaultMessage: 'Filter out this {value}',
     values: { value: columnId },