Corrected a minor issue in ai