fix wrong OrderStatus in BalanceTracker

jordantete · Jan 8, 2025 · 74826d0 · 74826d0
1 parent 617f96b
commit 74826d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/order_handling/balance_tracker.py b/core/order_handling/balance_tracker.py
@@ -166,8 +166,8 @@ def update_after_initial_purchase(self, initial_order: Order):
         Args:
             initial_order: The Order object containing details of the completed purchase.
         """
-        if initial_order.status != OrderStatus.FILLED:
-            raise ValueError(f"Order {initial_order.id} is not FILLED. Cannot update balances.")
+        if initial_order.status != OrderStatus.CLOSED:
+            raise ValueError(f"Order {initial_order.id} is not CLOSED. Cannot update balances.")
 
         total_cost = initial_order.filled * initial_order.average
         fee = self.fee_calculator.calculate_fee(initial_order.amount * initial_order.average)