diff --git a/src/io/multi_test.go b/src/io/multi_test.go index 7a24a8afc5..934a6ec785 100644 --- a/src/io/multi_test.go +++ b/src/io/multi_test.go @@ -332,9 +332,7 @@ func TestMultiReaderFreesExhaustedReaders(t *testing.T) { buf1 := bytes.NewReader([]byte("foo")) buf2 := bytes.NewReader([]byte("bar")) mr = MultiReader(buf1, buf2) - runtime.SetFinalizer(buf1, func(*bytes.Reader) { - close(closed) - }) + runtime.AddCleanup(buf1, func(ch chan struct{}) { close(ch) }, closed) }() buf := make([]byte, 4)