I just came up with an Idea!!
a new upgrade, Drone Repair.
When you get a limb chopped off and you have the upgrade, It uses up energy slowly and heals your limb or missing part!! I really think this would go with the gameplay well and could be implemented and improved by you.