Fix typo/logic error in previous commit