diff --git a/frontend/src/components/analytics/TransactionDistribution.tsx b/frontend/src/components/analytics/TransactionDistribution.tsx
index 2deb860..e440e09 100644
--- a/frontend/src/components/analytics/TransactionDistribution.tsx
+++ b/frontend/src/components/analytics/TransactionDistribution.tsx
@@ -50,15 +50,13 @@ export default function TransactionDistribution({
 
   // Create pie chart data from account balances
   const pieData: PieDataPoint[] = accounts.map((account, index) => {
-    const closingBalance = account.balances.find(
-      (balance) => balance.balance_type === "closingBooked"
-    );
+    const primaryBalance = account.balances?.[0]?.amount || 0;
     
     const colors = ["#3B82F6", "#10B981", "#F59E0B", "#EF4444", "#8B5CF6"];
     
     return {
       name: getAccountDisplayName(account),
-      value: closingBalance?.amount || 0,
+      value: primaryBalance,
       color: colors[index % colors.length],
     };
   });
diff --git a/frontend/src/routes/analytics.tsx b/frontend/src/routes/analytics.tsx
index 4b894e0..6b13643 100644
--- a/frontend/src/routes/analytics.tsx
+++ b/frontend/src/routes/analytics.tsx
@@ -53,10 +53,8 @@ function AnalyticsDashboard() {
   }
 
   const totalBalance = accounts?.reduce((sum, account) => {
-    const closingBalance = account.balances.find(
-      (balance) => balance.balance_type === "closingBooked"
-    );
-    return sum + (closingBalance?.amount || 0);
+    const primaryBalance = account.balances?.[0]?.amount || 0;
+    return sum + primaryBalance;
   }, 0) || 0;
 
   return (