Garmin edge 520 battery life issues, new head unit or replace the battery?

Equipment
11

The 520 was getting a bad wrap on the Garmin Forums around battery life. Mine declined a lot after a firmware update. In the end I went for a 1030. I would speak to Garmin and also check out the Garmin 520 forum.