Fixed some AI behaviour