Small fix to previous patch