battery fix

Change-Id: If774050f42aaa1a3c8cd47a9be96d3db4aa9cc5c