Update battery script with fix.