Minor fix in TestWS