From a44f7266c213bbb5d74137367d9a7ef98c3c495e Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Thu, 5 Jan 2017 12:32:44 -0800
Subject: [PATCH] Fix SaveRestoreState

---
 client/client_test.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/client/client_test.go b/client/client_test.go
index 4dfeb20cc..6c8bbe4f1 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -682,9 +682,12 @@ func TestClient_SaveRestoreState(t *testing.T) {
 	c2.allocLock.Lock()
 	for _, ar := range c2.allocs {
 		ar.Destroy()
-		<-ar.WaitCh()
 	}
 	c2.allocLock.Unlock()
+
+	for _, ar := range c2.allocs {
+		<-ar.WaitCh()
+	}
 }
 
 func TestClient_Init(t *testing.T) {