Fix typo #12
Update test cases for consistency #12