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
 	}