Re-enable update_stats in run