Small fix to previous