Fix AI bug