Fixing Xiaomi Piston 2 support

This needs the kernel changes I sent into separate pull request: https://github.com/M1cha/android_kernel_xiaomi_aries/pull/2