Xiaomi Says It's Working To Solve The Mi A1 Battery Issue