From 34b386b9447434d2f2019e2c5f45e351215eb294 Mon Sep 17 00:00:00 2001
From: Tolga Ceylan <tolga.ceylan@gmail.com>
Date: Fri, 6 Apr 2018 18:02:51 -0700
Subject: [PATCH] fn: lb_agent wait group unused, added to Close (#924)

---
 api/agent/lb_agent.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/api/agent/lb_agent.go b/api/agent/lb_agent.go
index 76a0099b6..be14ce5c1 100644
--- a/api/agent/lb_agent.go
+++ b/api/agent/lb_agent.go
@@ -138,6 +138,7 @@ func (a *lbAgent) Close() error {
 
 	a.rp.Shutdown(ctx)
 	err := a.delegatedAgent.Close()
+	a.wg.Wait()
 	if err != nil {
 		return err
 	}