From 3205e69e11925fc9f3511640f4e5b28eea35fd34 Mon Sep 17 00:00:00 2001
From: jonesmarvin8 <83104039+jonesmarvin8@users.noreply.github.com>
Date: Mon, 26 Jan 2026 15:15:04 -0500
Subject: [PATCH] Update programs/amm/src/new_definition.rs

Co-authored-by: Daniil Polyakov <arjentix@gmail.com>
---
 programs/amm/src/new_definition.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/programs/amm/src/new_definition.rs b/programs/amm/src/new_definition.rs
index ac62b95c..ec9efc94 100644
--- a/programs/amm/src/new_definition.rs
+++ b/programs/amm/src/new_definition.rs
@@ -16,8 +16,8 @@ pub fn new_definition(
     user_holding_a: AccountWithMetadata,
     user_holding_b: AccountWithMetadata,
     user_holding_lp: AccountWithMetadata,
-    token_a_amount: u128,
-    token_b_amount: u128,
+    token_a_amount: NonZeroU128,
+    token_b_amount: NonZeroU128,
     amm_program_id: ProgramId,
 ) -> (Vec<AccountPostState>, Vec<ChainedCall>) {
     // Prevents pool constant coefficient (k) from being 0.