From 7b160b7d0bb15c0821ff9b18a60e7c538b0d5449 Mon Sep 17 00:00:00 2001
From: Will Chen <willchen90@gmail.com>
Date: Fri, 19 Sep 2025 10:16:53 -0700
Subject: [PATCH] fix pro banners (#1331)

---
 src/components/ProBanner.tsx | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/components/ProBanner.tsx b/src/components/ProBanner.tsx
index 1b0c21c..d0b96d1 100644
--- a/src/components/ProBanner.tsx
+++ b/src/components/ProBanner.tsx
@@ -13,15 +13,16 @@ import { useUserBudgetInfo } from "@/hooks/useUserBudgetInfo";
 export function ProBanner() {
   const { settings } = useSettings();
   const { userBudget } = useUserBudgetInfo();
-  if (settings?.enableDyadPro || userBudget) {
-    return null;
-  }
 
   const [selectedBanner] = useState<"ai" | "smart" | "turbo">(() => {
     const options = ["ai", "smart", "turbo"] as const;
     return options[Math.floor(Math.random() * options.length)];
   });
 
+  if (settings?.enableDyadPro || userBudget) {
+    return null;
+  }
+
   return (
     <div className="mt-6 max-w-2xl mx-auto">
       {selectedBanner === "ai" ? (