Fix AI not found bug