From f619db297f0299e882f6b44381a06b0790f658d3 Mon Sep 17 00:00:00 2001
From: Danielle Tomlinson <dani@hashicorp.com>
Date: Tue, 18 Dec 2018 23:38:28 +0100
Subject: [PATCH] client: Update tests for async destroy

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

diff --git a/client/client_test.go b/client/client_test.go
index e5e68ed1c..f81cfe93f 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -642,7 +642,7 @@ func TestClient_SaveRestoreState(t *testing.T) {
 	}
 
 	for _, ar := range c2.getAllocRunners() {
-		<-ar.WaitCh()
+		<-ar.DestroyCh()
 	}
 }
 
@@ -780,7 +780,7 @@ func TestClient_BlockedAllocations(t *testing.T) {
 	}
 
 	for _, ar := range c1.getAllocRunners() {
-		<-ar.WaitCh()
+		<-ar.DestroyCh()
 	}
 }