Xiaomi air purifier, states incorrect

The air purifier is often at state unknown, which is an issue if you want to have it on for x minutes and it is interrupted by unknown.

Is there an alternative to this plugin or is there a fix in HA?