diff --git a/services/web/app/src/Features/Subscription/PaymentProviderEntities.js b/services/web/app/src/Features/Subscription/PaymentProviderEntities.js
index 7d1d7a2226..9485eb0656 100644
--- a/services/web/app/src/Features/Subscription/PaymentProviderEntities.js
+++ b/services/web/app/src/Features/Subscription/PaymentProviderEntities.js
@@ -13,7 +13,7 @@
  * @property {number} subtotal
  * @property {number} discount
  * @property {number} tax
- * @property {boolean} isAiAssist
+ * @property {boolean} [isAiAssist]
  */
 
 const OError = require('@overleaf/o-error')
diff --git a/services/web/frontend/js/features/subscription/components/preview-subscription-change/root.tsx b/services/web/frontend/js/features/subscription/components/preview-subscription-change/root.tsx
index 7ebcfc301e..c54d10c9a6 100644
--- a/services/web/frontend/js/features/subscription/components/preview-subscription-change/root.tsx
+++ b/services/web/frontend/js/features/subscription/components/preview-subscription-change/root.tsx
@@ -39,6 +39,7 @@ function PreviewSubscriptionChange() {
     (item, index, arr) => {
       if (!item.isAiAssist) return true
 
+      // TODO: this can be removed when all subscriptions are using Stripe
       const isCanceledByAnotherItem = arr.some(
         (otherItem, otherIndex) =>
           otherIndex !== index &&
diff --git a/services/web/types/subscription/subscription-change-preview.ts b/services/web/types/subscription/subscription-change-preview.ts
index 4821b650e7..c6fd717370 100644
--- a/services/web/types/subscription/subscription-change-preview.ts
+++ b/services/web/types/subscription/subscription-change-preview.ts
@@ -9,7 +9,7 @@ export type ImmediateCharge = {
     subtotal: number
     discount: number
     tax: number
-    isAiAssist: boolean
+    isAiAssist?: boolean
   }[]
 }