Fix minor coverage problem

Fix minor coverage problem.