Fix ws routes