correct typo
add unittest