Change web socket test to tear down connection