grammaticality - "How much water do you take a bath with?" — Is this sentence correct?

I corrected the student, saying that he should write "How much water do you use to take a bath?" because his sentence seemed unnatural to me. Do you consider it correct? Would you use it?