Fix flake hopefully

Author: spalmurray

src/pages/PlanPage/subRoutes/UpgradePlanPage/UpgradeForm/Controllers/ProPlanController/BillingOptions/BillingOptions.test.tsx

@@ -155,7 +155,7 @@ describe('BillingOptions', () => {
 
         const annualBtn = await screen.findByRole('button', { name: 'Annual' })
         expect(annualBtn).toBeInTheDocument()
-        expect(annualBtn).toHaveClass('bg-ds-primary-base')
+        await waitFor(() => expect(annualBtn).toHaveClass('bg-ds-primary-base'))
 
         const monthlyBtn = await screen.findByRole('button', {
           name: 'Monthly',

src/pages/PlanPage/subRoutes/UpgradePlanPage/UpgradeForm/Controllers/TeamPlanController/BillingOptions/BillingOptions.test.tsx

@@ -138,7 +138,7 @@ describe('BillingOptions', () => {
 
         const annualBtn = await screen.findByRole('button', { name: 'Annual' })
         expect(annualBtn).toBeInTheDocument()
-        expect(annualBtn).toHaveClass('bg-ds-primary-base')
+        await waitFor(() => expect(annualBtn).toHaveClass('bg-ds-primary-base'))
 
         const monthlyBtn = await screen.findByRole('button', {
           name: 'Monthly',