Fixed typo which was messing up tests