From 578c24b17f10142031689f8b45429ee895c12333 Mon Sep 17 00:00:00 2001
From: Anton <43762166+smypmsa@users.noreply.github.com>
Date: Sun, 15 Jun 2025 20:49:40 +0200
Subject: [PATCH] Update src/trading/buyer.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
---
 src/trading/buyer.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/trading/buyer.py b/src/trading/buyer.py
index 5e0eccb..f40a2cb 100644
--- a/src/trading/buyer.py
+++ b/src/trading/buyer.py
@@ -341,6 +341,10 @@ class TokenBuyer(Trader):
         if tokens_received_raw <= 0:
             logger.warning("Token balance search failed. Using fallback from EXTREME_FAST estimate.")
             # Fallback: use the amount we know we bought
-            return self.extreme_fast_token_amount if self.extreme_fast_mode else 20.0
-            
+            if self.extreme_fast_mode and self.extreme_fast_token_amount > 0:
+                return self.extreme_fast_token_amount
+            else:
+                logger.error("Cannot determine tokens received from transaction")
+                return 0.0
+
         return tokens_received_raw / 10**TOKEN_DECIMALS
\ No newline at end of file