Rewrite web socket url as well

Signed-off-by: Ryan Benoit <rdbenoit@me.com>