Fix AI count tracking.