You now can find generic M12 battery for sale now at a very good price. No Milwaukee label, but other wise looks like the original battery
In particular Milwaukee generic M12 battery.
Seller says he has a Milwaukee 2404 M12 equivalent battery for sale, which should be a 4.0 amp battery at...