Fix test_client.py