fix spot liab bug

drift-labs · Nov 15, 2022 · 68e21cf · 68e21cf
1 parent b9b52c0
commit 68e21cf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/driftpy/clearing_house_user.py b/src/driftpy/clearing_house_user.py
@@ -217,7 +217,7 @@ async def get_spot_market_liability(
 
             if worst_case_token_amount < 0:
                 baa_value = get_spot_liability_value(
-                    worst_case_token_amount,
+                    abs(worst_case_token_amount),
                     oracle_data,
                     spot_market,
                     margin_category,
@@ -226,12 +226,12 @@ async def get_spot_market_liability(
                 )
                 total_liability += baa_value
 
-            if worst_case_quote_amount > 0:
+            if worst_case_quote_amount < 0:
                 weight = SPOT_WEIGHT_PRECISION
                 if margin_category == MarginCategory.INITIAL:
                     weight = max(weight, user.max_margin_ratio)
                 weighted_value = (
-                    worst_case_quote_amount * weight / SPOT_WEIGHT_PRECISION
+                    abs(worst_case_quote_amount) * weight / SPOT_WEIGHT_PRECISION
                 )
                 total_liability += weighted_value