Fix tests

rytilahti · Jan 22, 2019 · be18a22 · be18a22
1 parent 0d5f956
commit be18a22
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/miio/tests/test_airqualitymonitor.py b/miio/tests/test_airqualitymonitor.py
@@ -3,12 +3,13 @@
 import pytest
 
 from miio import AirQualityMonitor
-from miio.airqualitymonitor import AirQualityMonitorStatus
+from miio.airqualitymonitor import AirQualityMonitorStatus, MODEL_AIRQUALITYMONITOR_V1
 from .dummies import DummyDevice
 
 
 class DummyAirQualityMonitor(DummyDevice, AirQualityMonitor):
     def __init__(self, *args, **kwargs):
+        self.model = MODEL_AIRQUALITYMONITOR_V1
         self.state = {
             'power': 'on',
             'aqi': 34,