Small fix for the previous commit