Small fix from the previous commit