diff --git a/custom_components/houseplan/websocket_api.py b/custom_components/houseplan/websocket_api.py
index b5bb00c..04d87ac 100755
--- a/custom_components/houseplan/websocket_api.py
+++ b/custom_components/houseplan/websocket_api.py
@@ -16,6 +16,7 @@ from homeassistant.components import websocket_api
 from homeassistant.core import HomeAssistant, callback
 
 from .const import (
+    DOMAIN,
     CONF_ADMIN_ONLY, DEFAULT_CONFIG,
     CONTENT_URL, FILES_DIR, MAX_PLANS_BYTES, MAX_PLANS_FILES, MAX_PLANS_LISTED,
     MAX_SIGN_PATHS,