Adjusting Python client unit test to match changes