Fix coverage generation