Add second wait group to make sure that reconnecting and receiving don't conflict.