Fixed a typo and updated tests