Updating the tests and fixing a typo