diff --git a/fastapi/websockets.py b/fastapi/websockets.py
index bed672acff5f25..55a4ac4a1a9187 100644
--- a/fastapi/websockets.py
+++ b/fastapi/websockets.py
@@ -1,2 +1,3 @@
 from starlette.websockets import WebSocket as WebSocket  # noqa
 from starlette.websockets import WebSocketDisconnect as WebSocketDisconnect  # noqa
+from starlette.websockets import WebSocketState as WebSocketState  # noqa