Fix coverage collection