Fix battery drain