Improve test coverage
Add test