Minor tweak to match current featureset