diff --git a/test/function_test.go b/test/function_test.go
index 191caf12..3196a9c6 100644
--- a/test/function_test.go
+++ b/test/function_test.go
@@ -33,7 +33,7 @@ var (
 func TestFunctions(t *testing.T) {
 	t.Run(runtime.FuncForPC(reflect.ValueOf(PingPodIP).Pointer()).Name(), PingPodIP)
 	t.Run(runtime.FuncForPC(reflect.ValueOf(UDP).Pointer()).Name(), UDP)
-	t.Run("cancel", func(t *testing.T) { cancelFunc() })
+	t.Cleanup(cancelFunc)
 }
 
 func PingPodIP(t *testing.T) {