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()
 	}
 }