Fix ws error