websock fix
move to demo code