Update test_stats_client.py