Update tests to pass client_args rather than headers