Fix coverage tests