diff --git a/tests/test_client_legacy.py b/tests/test_client_legacy.py
index 9598acaa94..f74ab2c777 100644
--- a/tests/test_client_legacy.py
+++ b/tests/test_client_legacy.py
@@ -591,7 +591,7 @@ def has_model_endpoint_type(models: List["LLMConfig"], target_type: str) -> bool
         assert has_model_endpoint_type(models, "anthropic")
 
 
-def test_shared_blocks(client: Union[LocalClient, RESTClient], agent: AgentState):
+def test_shared_blocks(mock_e2b_api_key_none, client: Union[LocalClient, RESTClient], agent: AgentState):
     # _reset_config()
 
     # create a block