Improve test coverage
2 files changed