fix ai for new version