diff --git a/home-assistant/custom_components/samsung_multi_room/media_player.py b/home-assistant/custom_components/samsung_multi_room/media_player.py
index 851aa30..9cdae30 100644
--- a/home-assistant/custom_components/samsung_multi_room/media_player.py
+++ b/home-assistant/custom_components/samsung_multi_room/media_player.py
@@ -27,7 +27,7 @@
 )
 
 from homeassistant.components.media_player import (
-  MediaPlayerDevice
+  MediaPlayerEntity
 )
 
 from homeassistant.const import (
@@ -125,7 +125,7 @@ async def get_source(self):
   async def set_source(self, source):
     return await self._exec_set('SetFunc', 'function', source)
 
-class MultiRoomDevice(MediaPlayerDevice):
+class MultiRoomDevice(MediaPlayerEntity):
   def __init__(self, name, max_volume, api):
     _LOGGER.info('Initializing MultiRoomDevice')
     self._name = name