diff --git a/client/client_test.go b/client/client_test.go
index 69f77caff..0455ca3fa 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -899,7 +899,7 @@ func TestClient_UnarchiveAllocDir(t *testing.T) {
 
 	rc := ioutil.NopCloser(buf)
 
-	c1.migratingAllocs["123"] = make(chan struct{})
+	c1.migratingAllocs["123"] = newMigrateAllocCtrl()
 	if err := c1.unarchiveAllocDir(rc, "123", dir1); err != nil {
 		t.Fatalf("err: %v", err)
 	}