fix typo
add more tests