AI bugfixes