Fix class name typo
Add test cases