diff --git a/libs/ui-lib/lib/cim/components/Agent/tableUtils.tsx b/libs/ui-lib/lib/cim/components/Agent/tableUtils.tsx
index f95fde7c11..e9e00d2c3f 100644
--- a/libs/ui-lib/lib/cim/components/Agent/tableUtils.tsx
+++ b/libs/ui-lib/lib/cim/components/Agent/tableUtils.tsx
@@ -583,7 +583,7 @@ export const useAgentsFilter = ({
 }: UseAgentsFilterArgs) => {
   const [hostnameFilter, setHostnameFilter] = React.useState<string>();
   const [statusFilter, setStatusFilter] = React.useState<string[]>([]);
-  const statusCount = Object.keys(agentStatus).reduce((acc, curr) => {
+  const statusCount = Object.keys(agentStatuses).reduce((acc, curr) => {
     acc[agentStatuses[curr].title] = 0;
     return acc;
   }, {} as StatusCount);
diff --git a/libs/ui-lib/lib/cim/components/InfraEnv/InfraTableToolbar.tsx b/libs/ui-lib/lib/cim/components/InfraEnv/InfraTableToolbar.tsx
index cecbf454f6..646c7c9762 100644
--- a/libs/ui-lib/lib/cim/components/InfraEnv/InfraTableToolbar.tsx
+++ b/libs/ui-lib/lib/cim/components/InfraEnv/InfraTableToolbar.tsx
@@ -35,19 +35,19 @@ const getStatusesForFilter = (statuses: HostStatus<string>) => {
   );
 
   sortedStatuses.forEach((status) => {
-    const { category, key } = statuses[status];
+    const { category, title } = statuses[status];
     if (filterStatuses[category]) {
-      if (filterStatuses[category][key]) {
-        filterStatuses[category][key].push(status);
+      if (filterStatuses[category][title]) {
+        filterStatuses[category][title].push(status);
       } else {
         filterStatuses[category] = {
           ...filterStatuses[category],
-          [key]: [status],
+          [title]: [status],
         };
       }
     } else {
       filterStatuses[category] = {
-        [key]: [status],
+        [title]: [status],
       };
     }
   });