diff --git a/beamer/tests/agent/test_request.py b/beamer/tests/agent/test_request.py
index baf2b410f..e2aa63398 100644
--- a/beamer/tests/agent/test_request.py
+++ b/beamer/tests/agent/test_request.py
@@ -377,7 +377,7 @@ def test_event_fetcher_progress_without_events(agent, direction):
         ape.chain.mine(1)
         with Sleeper(5) as sleeper:
             while (
-                agent.get_context(direction).latest_blocks[ape.chain.chain_id].number
+                agent.get_context(direction).latest_blocks[ape.chain.chain_id]["number"]
                 != ape.chain.blocks[-1].number
             ):
                 sleeper.sleep(0.1)
@@ -400,7 +400,7 @@ def test_handling_reorgs(config, direction, token, request_manager):
     with Sleeper(1) as sleeper:
         while (
             ChainId(ape.chain.chain_id) not in agent.get_context(direction).latest_blocks
-            or agent.get_context(direction).latest_blocks[ChainId(ape.chain.chain_id)].number
+            or agent.get_context(direction).latest_blocks[ChainId(ape.chain.chain_id)]["number"]
             != expected_block_number
         ):
             sleeper.sleep(0.1)