diff --git a/invokeai/frontend/web/src/features/controlLayers/store/selectors.ts b/invokeai/frontend/web/src/features/controlLayers/store/selectors.ts
index 44ac9f32b8..2269f393f3 100644
--- a/invokeai/frontend/web/src/features/controlLayers/store/selectors.ts
+++ b/invokeai/frontend/web/src/features/controlLayers/store/selectors.ts
@@ -1,17 +1,8 @@
 import { createSelector } from '@reduxjs/toolkit';
-import { selectControlAdaptersV2Slice } from 'features/controlLayers/store/controlAdaptersSlice';
-import { selectIPAdaptersSlice } from 'features/controlLayers/store/ipAdaptersSlice';
-import { selectLayersSlice } from 'features/controlLayers/store/layersSlice';
-import { selectRegionalGuidanceSlice } from 'features/controlLayers/store/regionalGuidanceSlice';
+import { selectCanvasSlice } from 'features/canvas/store/canvasSlice';
 
-export const selectEntityCount = createSelector(
-  selectRegionalGuidanceSlice,
-  selectControlAdaptersV2Slice,
-  selectIPAdaptersSlice,
-  selectLayersSlice,
-  (rgState, caState, ipaState, layersState) => {
-    return (
-      rgState.regions.length + caState.controlAdapters.length + ipaState.ipAdapters.length + layersState.layers.length
-    );
-  }
-);
+export const selectEntityCount = createSelector(selectCanvasSlice, (canvasV2) => {
+  return (
+    canvasV2.regions.length + canvasV2.controlAdapters.length + canvasV2.ipAdapters.length + canvasV2.layers.length
+  );
+});