[FIXED] Final AI bugs