adding tests / fixing typo