The sentence is understandable as is. It's not unnatural to me.
I don't think it sounds entirely unnatural. The only thing is we don't really say "to do treatment". We use "to treat" or to "provide treatment". Consequently, I wouldn't head for "treatment being done".