Fix websocket issue on heroku