Fixed websocket bug

Signed-off-by: Vishal Rana <vr@labstack.com>