diff --git a/napalm_sonos/sonos_driver.py b/napalm_sonos/sonos_driver.py
index c0868d4..92e991e 100644
--- a/napalm_sonos/sonos_driver.py
+++ b/napalm_sonos/sonos_driver.py
@@ -6,7 +6,7 @@ from typing import Any
 
 from napalm.base.exceptions import ConnectionException
 
-from napalm_device_types import AccessPointDriver, FingerprintRule, PortSpec
+from napalm_device_types import FingerprintRule, MediaDriver, PortSpec
 from napalm_device_types.models import DeviceActionResultDict
 
 """NAPALM driver for Sonos speakers.
@@ -19,7 +19,7 @@ media-playback controls.
 _VOLUME_RE = re.compile(r"^\d+$")
 
 
-class SonosDriver(AccessPointDriver):
+class SonosDriver(MediaDriver):
     """NAPALM driver for controlling Sonos speakers.
 
     Implements the standard NAPALM interface plus media-playback methods