diff --git a/src/plugin-hooks/useRowSelect.js b/src/plugin-hooks/useRowSelect.js
index b697070..6b6de24 100644
--- a/src/plugin-hooks/useRowSelect.js
+++ b/src/plugin-hooks/useRowSelect.js
@@ -104,7 +104,7 @@ function reducer(state, action, previousState, instance) {
         selectedRowIds[rowId] = true
       })
     } else {
-      Object.keys(flatRowsById).forEach(rowId => {
+      Object.keys(nonGroupedRowsById).forEach(rowId => {
         delete selectedRowIds[rowId]
       })
     }