AI Bug fix