Increasing code coverage in tests