Add tests
Fix typo