Minor cleanup in AI system