AI bugfix