i2c-i801: Restore the device state before leaving