diff --git a/tests/local_testing/test_custom_callback_input.py b/tests/local_testing/test_custom_callback_input.py
index d18668ebf1c..62875a5a115 100644
--- a/tests/local_testing/test_custom_callback_input.py
+++ b/tests/local_testing/test_custom_callback_input.py
@@ -1559,7 +1559,7 @@ def test_logging_standard_payload_llm_headers(stream):
                 continue
 
         time.sleep(2)
-        mock_client.assert_called_once()
+        mock_client.assert_called()
 
         standard_logging_object: StandardLoggingPayload = mock_client.call_args.kwargs[
             "kwargs"