Add a flag for xiaomi