Fix battery representation