Fixed minor error of the previous commit