Change logic to identify ws client with token.