Small fix to previous one