Polish (minor)
Add WebSocketHandlerDecorator