AI bug fixed