Fix coverage issue