More AI fixes