Fixed test with use_ws