Fix typo
Improve test coverage.