Small fix on a feature check