Small fix related to the previous commit