diff --git a/tests/test_reward.py b/tests/test_reward.py
index dcdd8d82..9387d0a4 100644
--- a/tests/test_reward.py
+++ b/tests/test_reward.py
@@ -16,10 +16,10 @@ def test_rewards_are_being_penalised_at_each_step_function():
 
     """
     On different steps (of the 13 in total) these are the following rewards for config_6 which are activated:
-        File System State: goodShouldBeCorrupt = 5 (Step 3)
-        Hardware State: onShouldBeOff = -2 (Step 5)
-        Service State: goodShouldBeCompromised = 5 (Step 7)
-        Operating System State (Software State): goodShouldBeCompromised = 5 (Step 10)
+        File System State: goodShouldBeCorrupt = 5 (between Steps 1 & 3)
+        Hardware State: onShouldBeOff = -2 (between Steps 4 & 6)
+        Service State: goodShouldBeCompromised = 5 (between Steps 7 & 9)
+        Operating System State (Software State): goodShouldBeCompromised = 5 (between Steps 10 & 12)
 
     Total Reward: -2 - 2 + 5 + 5 + 5 + 5 + 5 + 5 = 26
     Step Count: 13