Fixed an issue with the battery integration test.