Code coverage fix