fix test_stats.py