minor ai fixes