diff --git a/fn/context_guard_test.go b/fn/context_guard_test.go
index e13cba5dc..76d63a50f 100644
--- a/fn/context_guard_test.go
+++ b/fn/context_guard_test.go
@@ -298,6 +298,12 @@ func TestContextGuard(t *testing.T) {
 		case <-time.After(time.Second):
 			t.Fatalf("timeout")
 		}
+
+		// Cancel the context.
+		cancel()
+
+		// Make sure wg's counter gets to 0 eventually.
+		g.WgWait()
 	})
 
 	// Test that if we add the CustomTimeoutCGOpt option, then the context