diff --git a/zha/websocket/server/__main__.py b/zha/websocket/server/__main__.py
index ac80c2721..42847319f 100644
--- a/zha/websocket/server/__main__.py
+++ b/zha/websocket/server/__main__.py
@@ -9,7 +9,7 @@
 from pathlib import Path
 
 from zha.application.gateway import WebSocketServerGateway
-from zha.application.helpers import (
+from zha.application.model import (
     WebsocketClientConfiguration,
     WebsocketServerConfiguration,
     ZHAConfiguration,