Fix typo and add test