diff --git a/tests/const.py b/tests/const.py
index 4efa192..db73d52 100644
--- a/tests/const.py
+++ b/tests/const.py
@@ -7,12 +7,12 @@
 from custom_components.sunspec.const import CONF_SLAVE_ID
 
 MOCK_SETTINGS_PREFIX = {
-    CONF_ENABLED_MODELS: ["160"],
+    CONF_ENABLED_MODELS: [160],
     CONF_PREFIX: "test",
     CONF_SCAN_INTERVAL: 10,
 }
-MOCK_SETTINGS = {CONF_ENABLED_MODELS: ["103", "160"], CONF_SCAN_INTERVAL: 10}
-MOCK_SETTINGS_MM = {CONF_ENABLED_MODELS: ["701"], CONF_SCAN_INTERVAL: 10}
+MOCK_SETTINGS = {CONF_ENABLED_MODELS: [103, 160], CONF_SCAN_INTERVAL: 10}
+MOCK_SETTINGS_MM = {CONF_ENABLED_MODELS: [701], CONF_SCAN_INTERVAL: 10}
 MOCK_CONFIG_STEP_1 = {CONF_HOST: "test_host", CONF_PORT: 123, CONF_SLAVE_ID: 1}
 MOCK_CONFIG = {
     CONF_HOST: "test_host",