diff --git a/tests/components/sabnzbd/test_init.py b/tests/components/sabnzbd/test_init.py
index 9bdef4119d0..f140c332778 100644
--- a/tests/components/sabnzbd/test_init.py
+++ b/tests/components/sabnzbd/test_init.py
@@ -3,7 +3,7 @@ from unittest.mock import patch
 
 import pytest
 
-from homeassistant.components.sabnzbd import DEFAULT_NAME, DOMAIN, SENSOR_KEYS
+from homeassistant.components.sabnzbd import DEFAULT_NAME, DOMAIN, OLD_SENSOR_KEYS
 from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
 from homeassistant.const import CONF_API_KEY, CONF_NAME, CONF_URL
 from homeassistant.helpers.device_registry import DeviceEntryType
@@ -54,7 +54,7 @@ async def test_unique_id_migrate(hass, device_registry, entity_registry):
 
     entity_id_sensor_key = []
 
-    for sensor_key in SENSOR_KEYS:
+    for sensor_key in OLD_SENSOR_KEYS:
         mock_entity_id = f"{SENSOR_DOMAIN}.{DOMAIN}_{sensor_key}"
         entity_registry.async_get_or_create(
             SENSOR_DOMAIN,