Minor fix in AI