Small fix to previous commit