code coverage fix