Fix ws handler call