test coverage fix