diff --git a/platforms/HomeAssistant.js b/platforms/HomeAssistant.js
index 4c12a40..3929b3e 100644
--- a/platforms/HomeAssistant.js
+++ b/platforms/HomeAssistant.js
@@ -399,9 +399,9 @@ HomeAssistantMediaPlayer.prototype = {
 }
 
 
-function HomeAssistantSwitch(log, data, client) {
+function HomeAssistantSwitch(log, data, client, type) {
   // device info
-  this.domain = "switch"
+  this.domain = type || "switch"
   this.data = data
   this.entity_id = data.entity_id
   if (data.attributes && data.attributes.friendly_name) {