Fix typo
Fix typo
Add more tests