Fixed incorrect code coverage