Fix an error from the previous commit