Correct AI bug