Fixed code coverage