battery replacement - best options?

Hello I am looking for a battery replacement for the Z1E Rog Ally, as I have ran through the one i have and it is on 80% health (probably from keeping it at 100 too much at high temps). I play mostly plugged in, but am interested if there's any higher watt battery's that fit in the console.