small fix to ai