Fix coverage issue here