Increasing code coverage in unit tests