diff --git a/miio/tests/test_airqualitymonitor.py b/miio/tests/test_airqualitymonitor.py
index 11fe312e2..328484b85 100644
--- 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,