Fix errors due to multiple output from one AI