From f96e6b83d69d41e1d94a19b88c68be7b08d38876 Mon Sep 17 00:00:00 2001
From: Brezo Cordero <8002881+brezocordero@users.noreply.github.com>
Date: Wed, 20 Mar 2024 14:44:00 -0500
Subject: [PATCH] Also show upgrade message and link if key is valid without
 premium

[MAILPOET-5938]
---
 .../premium-banner-with-upgrade.tsx                  | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/mailpoet/assets/js/src/common/premium-banner-with-upgrade/premium-banner-with-upgrade.tsx b/mailpoet/assets/js/src/common/premium-banner-with-upgrade/premium-banner-with-upgrade.tsx
index 194fe3df0c..02d756d010 100644
--- a/mailpoet/assets/js/src/common/premium-banner-with-upgrade/premium-banner-with-upgrade.tsx
+++ b/mailpoet/assets/js/src/common/premium-banner-with-upgrade/premium-banner-with-upgrade.tsx
@@ -86,15 +86,19 @@ export function PremiumBannerWithUpgrade({
 
     ctaButton = getCtaButton(__('Upgrade your plan', 'mailpoet'), link);
   } else if (
-    hasValidPremiumKey &&
-    capabilityName &&
-    MailPoet.capabilities[capabilityName].isRestricted
+    (hasValidApiKey && !hasValidPremiumKey) || // ex. Starter plan
+    (hasValidPremiumKey &&
+      capabilityName &&
+      MailPoet.capabilities[capabilityName].isRestricted)
   ) {
     title = __('Upgrade your plan', 'mailpoet');
     bannerMessage = message;
+    const upgradeParams = capabilityName
+      ? { capability: capabilityName, s: subscribersCount }
+      : {};
     const link = MailPoet.MailPoetComUrlFactory.getUpgradeUrl(
       pluginPartialKey,
-      { capability: capabilityName, s: subscribersCount },
+      upgradeParams,
     );
     ctaButton = getCtaButton(__('Upgrade', 'mailpoet'), link);
   } else {