diff --git a/wideq/client.py b/wideq/client.py
index b7d68b7..76428f3 100644
--- a/wideq/client.py
+++ b/wideq/client.py
@@ -152,7 +152,7 @@ class Client(object):
             return None
         classes = util.device_classes()
         if deviceInfo.type in classes:
-            return classes.get(deviceInfo.type)(self, deviceInfo)
+            return classes[deviceInfo.type](self, deviceInfo)
         LOGGER.debug('No specific subclass for deviceType %s, using default',
                      deviceInfo.type)
         return Device(self, deviceInfo)