diff --git a/frontend/pages/policies/PolicyPage/components/PolicyForm/PolicyForm.tsx b/frontend/pages/policies/PolicyPage/components/PolicyForm/PolicyForm.tsx index e515540027..e03d46c9ba 100644 --- a/frontend/pages/policies/PolicyPage/components/PolicyForm/PolicyForm.tsx +++ b/frontend/pages/policies/PolicyPage/components/PolicyForm/PolicyForm.tsx @@ -227,14 +227,17 @@ const PolicyForm = ({ if (!isEditMode) { setIsNewPolicyModalOpen(true); } else { - onUpdate({ + const payload: IPolicyFormData = { name: lastEditedQueryName, description: lastEditedQueryDescription, query: lastEditedQueryBody, resolution: lastEditedQueryResolution, - critical: lastEditedQueryCritical, platform: newPlatformString, - }); + }; + if (isPremiumTier) { + payload.critical = lastEditedQueryCritical; + } + onUpdate(payload); } setIsEditingName(false);