Battery draining fast after android m update in mi4w

[B]hey,guys I recently got marshmallow update on my mi4w.
Everything was looking good,After update but battery draining too fast mostly when using wifi or hotspot.
Please fix it. Kitkat rom was best at battry saving but not marshmallow.