Fix tests for stats