diff --git a/src/components/modals/StakeCAP.svelte b/src/components/modals/StakeCAP.svelte
index 2e4bf18..35f0769 100644
--- a/src/components/modals/StakeCAP.svelte
+++ b/src/components/modals/StakeCAP.svelte
@@ -42,6 +42,10 @@
 		walletBalance = await getCAPWalletBalance();
 	}
 
+	function setMaxAmount() {
+		amount = walletBalance || 0;
+	}
+
 	checkAllowance();
 	getBalance();
 
@@ -69,7 +73,7 @@
 					value={formattedWalletBalance}
 					isClickable={true}
 					hasSemiPadding={true}
-					on:click={() => { amount = formattedWalletBalance; }}
+					on:click={setMaxAmount}
 				/>
 			</div>
 
diff --git a/src/components/modals/UnstakeCAP.svelte b/src/components/modals/UnstakeCAP.svelte
index 911532e..e1aceb8 100644
--- a/src/components/modals/UnstakeCAP.svelte
+++ b/src/components/modals/UnstakeCAP.svelte
@@ -33,6 +33,10 @@
 		focusInput('Amount');
 	});
 
+	function setMaxAmount() {
+		amount = $CAPStake || 0;
+	}
+
 
 </script>
 
@@ -53,7 +57,7 @@
 					value={formattedCAPStaked}
 					isClickable={true}
 					hasSemiPadding={true}
-					on:click={() => { amount = formattedCAPStaked; }}
+					on:click={setMaxAmount}
 				/>
 			</div>