Xiaomi India — replace the battery

I have replaced the battery in authorized mi service centre on 26/3/19... Due to swelling of battery on my mi max phone.. But the replaced battery has again is not working properly and the battery gets dead in 2 hrs... If i go to service centre they are rejecting the replacement... Hence i request mi india to take necessary action and replace the battery or else i will lodge complaint in the consumer court.