Fixed a bug in ai mode.