Added missing tests for 100% coverage
2 files changed