From 1c06dad425be69d9fa021790489bd03933d8f360 Mon Sep 17 00:00:00 2001
From: Adamantios <adamantios.zaras@valory.xyz>
Date: Mon, 13 May 2024 15:41:51 +0300
Subject: [PATCH] fix: correct the post-conditions for the benchmarking

---
 packages/valory/skills/decision_maker_abci/rounds.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/packages/valory/skills/decision_maker_abci/rounds.py b/packages/valory/skills/decision_maker_abci/rounds.py
index a71d7bc3d..61abad4c8 100644
--- a/packages/valory/skills/decision_maker_abci/rounds.py
+++ b/packages/valory/skills/decision_maker_abci/rounds.py
@@ -343,7 +343,8 @@ class DecisionMakerAbciApp(AbciApp[Event]):
         },
         BenchmarkingModeDisabledRound: set(),
         FinishedBenchmarkingRound: {
-            get_name(SynchronizedData.sampled_bet_index),
+            get_name(SynchronizedData.mocking_mode),
+            get_name(SynchronizedData.next_mock_data_row),
         },
         FinishedDecisionRequestRound: set(),
         FinishedSubscriptionRound: {
@@ -354,5 +355,8 @@ class DecisionMakerAbciApp(AbciApp[Event]):
         FinishedWithoutRedeemingRound: set(),
         RefillRequiredRound: set(),
         ImpossibleRound: set(),
-        BenchmarkingDoneRound: set(),
+        BenchmarkingDoneRound: {
+            get_name(SynchronizedData.mocking_mode),
+            get_name(SynchronizedData.next_mock_data_row),
+        },
     }