Fixing Xiaomi Piston 2 support (needs fixed kernel)

Change-Id: I20775620d42a21c375b82f094bdef1e3c9684269