fix typo
add more test cases