diff --git a/tests/test_client.py b/tests/test_client.py
index 532d8058..5db67157 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -347,7 +347,7 @@ def test_send_system_message(client: Union[LocalClient, RESTClient], agent: Agen
     assert send_system_message_response, "Sending message failed"
 
 
-def test_function_return_limit(mock_e2b_api_key_none, client: Union[LocalClient, RESTClient]):
+def test_function_return_limit(client: Union[LocalClient, RESTClient]):
     """Test to see if the function return limit works"""
 
     def big_return():