Fix coverage testing