diff --git a/packages/core/src/renderer/components/config-pod-disruption-budgets/__tests__/pod-distruption-budgets.test.tsx b/packages/core/src/renderer/components/config-pod-disruption-budgets/__tests__/pod-distruption-budgets.test.tsx
index c423962e68..33ab10f4aa 100644
--- a/packages/core/src/renderer/components/config-pod-disruption-budgets/__tests__/pod-distruption-budgets.test.tsx
+++ b/packages/core/src/renderer/components/config-pod-disruption-budgets/__tests__/pod-distruption-budgets.test.tsx
@@ -18,6 +18,9 @@ import { Cluster } from "../../../../common/cluster/cluster";
 import hostedClusterInjectable from "../../../cluster-frame-context/hosted-cluster.injectable";
 import userPreferencesStateInjectable from "../../../../features/user-preferences/common/state.injectable";
 import type { DiContainer } from "@ogre-tools/injectable";
+import { registerFeature } from "@k8slens/feature-core";
+import { runInAction } from "mobx";
+import { tableFeature } from "../../../library";
 
 describe("<PodDisruptionBudgets />", () => {
   let di: DiContainer;
@@ -69,6 +72,10 @@ describe("<PodDisruptionBudgets />", () => {
         }),
       } as any,
     }));
+
+    runInAction(() => {
+      registerFeature(di, tableFeature);
+    });
   });
 
   describe("PDB with minAvailable 0", () => {
diff --git a/packages/core/src/renderer/components/kube-object-list-layout/kube-object-list-layout.test.tsx b/packages/core/src/renderer/components/kube-object-list-layout/kube-object-list-layout.test.tsx
index ee52707837..9239310005 100644
--- a/packages/core/src/renderer/components/kube-object-list-layout/kube-object-list-layout.test.tsx
+++ b/packages/core/src/renderer/components/kube-object-list-layout/kube-object-list-layout.test.tsx
@@ -23,6 +23,9 @@ import type { PodStore } from "../workloads-pods/store";
 import { Cluster } from "../../../common/cluster/cluster";
 import isTableColumnHiddenInjectable from "../../../features/user-preferences/common/is-table-column-hidden.injectable";
 import { podListLayoutColumnInjectionToken } from "@k8slens/list-layout";
+import { registerFeature } from "@k8slens/feature-core";
+import { runInAction } from "mobx";
+import { tableFeature } from "../../library";
 
 describe("kube-object-list-layout", () => {
   let di: DiContainer;
@@ -54,6 +57,10 @@ describe("kube-object-list-layout", () => {
       get: () => ({}),
     }));
 
+    runInAction(() => {
+      registerFeature(di, tableFeature);
+    });
+
     podStore = di.inject(podStoreInjectable);
   });