Fix typo in feature step

I had the right code, then tested the inverse, then put it back wrongly. Ooops.