Update to use keep-alive as timeout for websockets