diff --git a/modules/caddyhttp/reverseproxy/streaming_test.go b/modules/caddyhttp/reverseproxy/streaming_test.go
index e6a3ce3a1..a7e3504b7 100644
--- a/modules/caddyhttp/reverseproxy/streaming_test.go
+++ b/modules/caddyhttp/reverseproxy/streaming_test.go
@@ -117,8 +117,8 @@ func TestHandlerCleanupLegacyModeClosesAllConnections(t *testing.T) {
 	ts := newTunnelState(caddy.Log(), 0)
 	connA := newTrackingReadWriteCloser()
 	connB := newTrackingReadWriteCloser()
-	ts.registerConnection(connA, nil, "a")
-	ts.registerConnection(connB, nil, "b")
+	ts.registerConnection(connA, nil, false, "a")
+	ts.registerConnection(connB, nil, false, "b")
 
 	h := &Handler{
 		tunnel:               ts,
@@ -136,7 +136,7 @@ func TestHandlerCleanupLegacyModeClosesAllConnections(t *testing.T) {
 func TestHandlerCleanupLegacyModeHonorsDelay(t *testing.T) {
 	ts := newTunnelState(caddy.Log(), 40*time.Millisecond)
 	conn := newTrackingReadWriteCloser()
-	ts.registerConnection(conn, nil, "a")
+	ts.registerConnection(conn, nil, false, "a")
 
 	h := &Handler{
 		tunnel:               ts,
@@ -175,8 +175,8 @@ func TestHandlerCleanupRetainModeClosesOnlyRemovedUpstreams(t *testing.T) {
 	ts := newTunnelState(caddy.Log(), 0)
 	connA := newTrackingReadWriteCloser()
 	connB := newTrackingReadWriteCloser()
-	ts.registerConnection(connA, nil, upstreamA)
-	ts.registerConnection(connB, nil, upstreamB)
+	ts.registerConnection(connA, nil, true, upstreamA)
+	ts.registerConnection(connB, nil, true, upstreamB)
 
 	h := &Handler{
 		tunnel:               ts,