Updates AI tracking to improve maintainability.