diff --git a/http2/transport_test.go b/http2/transport_test.go
index 322a4c4..7650046 100644
--- a/http2/transport_test.go
+++ b/http2/transport_test.go
@@ -934,6 +934,7 @@
 	}
 	ct.server = func() error {
 		ct.greet()
+		defer close(recvLen)
 		var buf bytes.Buffer
 		enc := hpack.NewEncoder(&buf)
 		var dataRecv int64
@@ -984,7 +985,7 @@
 				dataRecv += int64(dataLen)
 
 				if !closed && ((status != 200 && dataRecv > 0) ||
-					(status == 200 && dataRecv == bodySize)) {
+					(status == 200 && f.StreamEnded())) {
 					closed = true
 					if err := ct.fr.WriteData(f.StreamID, true, nil); err != nil {
 						return err