You can use a few smaller batteries like the original did.

You can use a smaller battery, or a bigger battery like I did

I ended up using a rechargeable 3.7V battery from a disposable vape (Elfbar 800).

This battery did not fit the puck case without modification, thus, I had to remove some plastics that held the original battery in place