Fix ai_performance module