diff --git a/host/host_test.go b/host/host_test.go
index 4425a44..5432f71 100644
--- a/host/host_test.go
+++ b/host/host_test.go
@@ -115,7 +115,7 @@ func TestVirtualization(t *testing.T) {
 	for i := 0; i < testCount; i++ {
 		go func(j int) {
 			system, role, err := Virtualization()
-			wg.Done()
+			defer wg.Done()
 			common.SkipIfNotImplementedErr(t, err)
 			assert.NoErrorf(t, err, "Virtualization() failed, %v", err)