Fixed typo and tests :/