Fix count => total_count