diff --git a/backend/alembic/versions/d88681a7b3d0_add_tier_to_plans.py b/backend/alembic/versions/d88681a7b3d0_add_tier_to_plans.py
index b33b6c4..7ef0c45 100644
--- a/backend/alembic/versions/d88681a7b3d0_add_tier_to_plans.py
+++ b/backend/alembic/versions/d88681a7b3d0_add_tier_to_plans.py
@@ -21,10 +21,12 @@
 def upgrade() -> None:
     """Upgrade schema."""
     # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_table('app_states')
-    op.drop_table('events')
-    op.drop_table('sessions')
-    op.drop_table('user_states')
+    conn = op.get_bind()
+    inspector = sa.inspect(conn)
+    existing = inspector.get_table_names()
+    for tbl in ('app_states', 'events', 'sessions', 'user_states'):
+        if tbl in existing:
+            op.drop_table(tbl)
     with op.batch_alter_table('plans', schema=None) as batch_op:
         batch_op.add_column(sa.Column('tier', sa.Integer(), server_default='0', nullable=False))
 
diff --git a/frontend/src/app/dashboard/widget-settings/page.tsx b/frontend/src/app/dashboard/widget-settings/page.tsx
index 154eb6c..cb697b0 100644
--- a/frontend/src/app/dashboard/widget-settings/page.tsx
+++ b/frontend/src/app/dashboard/widget-settings/page.tsx
@@ -79,10 +79,11 @@ export default function WidgetSettingsPage() {
       const res = await getBusinessProfile();
       if (res.status === 'success' && res.data) {
         setBusiness(res.data);
+        const d = res.data as unknown as Record<string, number | undefined>;
         setLimits({
-          allocated_messages_per_session: (res.data as Record<string, unknown>).allocated_messages_per_session,
-          allocated_daily_sessions: (res.data as Record<string, unknown>).allocated_daily_sessions,
-          allocated_whitelisted_domains: (res.data as Record<string, unknown>).allocated_whitelisted_domains,
+          allocated_messages_per_session: d.allocated_messages_per_session,
+          allocated_daily_sessions: d.allocated_daily_sessions,
+          allocated_whitelisted_domains: d.allocated_whitelisted_domains,
         });
       }
     } catch (e) { console.error(e); }