diff --git a/tests/test_arbitrage_integration.py b/tests/test_arbitrage_integration.py
index 8437a3c..9dd8b3a 100644
--- a/tests/test_arbitrage_integration.py
+++ b/tests/test_arbitrage_integration.py
@@ -67,7 +67,7 @@ def test_reverting_arbitrage(trace_classifier: TraceClassifier):
     assert len(swaps) == 38
 
     arbitrages = get_arbitrages(list(swaps))
-    assert len(arbitrages) == 21
+    assert len(arbitrages) == 4
 
     arbitrage_1 = [
         arb