diff --git a/src/types/onyx/Policy.ts b/src/types/onyx/Policy.ts
index 7d4c08374b81..5a1eddefa5dc 100644
--- a/src/types/onyx/Policy.ts
+++ b/src/types/onyx/Policy.ts
@@ -11,6 +11,7 @@ type Rate = {
     customUnitRateID?: string;
     errors?: OnyxCommon.Errors;
     pendingAction?: OnyxCommon.PendingAction;
+    enabled: boolean;
 };
 
 type Attributes = {
@@ -22,6 +23,8 @@ type CustomUnit = {
     customUnitID: string;
     attributes: Attributes;
     rates: Record<string, Rate>;
+    defaultCategory: string;
+    enabled: boolean;
     pendingAction?: OnyxCommon.PendingAction;
     errors?: OnyxCommon.Errors;
 };