More AI fixes.