diff --git a/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs b/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs
index 6ed5d5df629b..6a9656d9f730 100644
--- a/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs
+++ b/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http2/Http2ConnectionTests.cs
@@ -245,7 +245,7 @@ await ExpectAsync(Http2FrameType.HEADERS,
                 withStreamId: 1);
 
             // TriggerTick will trigger the stream to be returned to the pool so we can assert it
-            TriggerTick();
+            TriggerTick(_serviceContext.MockSystemClock.UtcNow);
 
             // Stream has been returned to the pool
             Assert.Equal(1, _connection.StreamPool.Count);
@@ -263,7 +263,7 @@ await ExpectAsync(Http2FrameType.HEADERS,
                 withStreamId: 3);
 
             // TriggerTick will trigger the stream to be returned to the pool so we can assert it
-            TriggerTick();
+            TriggerTick(_serviceContext.MockSystemClock.UtcNow);
 
             // Stream has been returned to the pool
             Assert.Equal(1, _connection.StreamPool.Count);
diff --git a/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http3/Http3ConnectionTests.cs b/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http3/Http3ConnectionTests.cs
index 614fdc569e79..b83f3b64deb4 100644
--- a/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http3/Http3ConnectionTests.cs
+++ b/src/Servers/Kestrel/test/InMemory.FunctionalTests/Http3/Http3ConnectionTests.cs
@@ -17,6 +17,7 @@
 using Microsoft.AspNetCore.Server.Kestrel.Core.Internal.Http3;
 using Microsoft.AspNetCore.Testing;
 using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Primitives;
 using Microsoft.Net.Http.Headers;
 using Xunit;
 using Http3SettingType = Microsoft.AspNetCore.Server.Kestrel.Core.Internal.Http3.Http3SettingType;