Fixing test coverage checks