fixed typo
added tests