Wrong device class for sensors

All sensors created by the integration gets device_class "safety", which to my understanding is used for binary sensors and not regular sensors. This removes the unit_of_measurement value for the sensors, and makes it hard to use the information from the sensor in home assistant. Removing the device_class from the sensors would be better. 

My hardware is 2 WAC510.