Add timeout to disconnect ws-client