diff --git a/gateway/handlers/logs.go b/gateway/handlers/logs.go
index 0a883fc6..7b784efa 100644
--- a/gateway/handlers/logs.go
+++ b/gateway/handlers/logs.go
@@ -8,6 +8,7 @@ import (
 	"net/url"
 	"os"
 	"strings"
+	"time"
 )
 
 const crlf = "\r\n"
diff --git a/gateway/handlers/logs_test.go b/gateway/handlers/logs_test.go
index 13bfd1a4..9c0f0ffd 100644
--- a/gateway/handlers/logs_test.go
+++ b/gateway/handlers/logs_test.go
@@ -8,6 +8,7 @@ import (
 	"net/http/httptest"
 	"net/url"
 	"testing"
+	"time"
 
 	"go.uber.org/goleak"
 )
@@ -42,7 +43,7 @@ func Test_logsProxyDoesNotLeakGoroutinesWhenProviderClosesConnection(t *testing.
 
 	logProviderURL, _ := url.Parse(mockLogsUpstreamEndpoint.URL)
 
-	logHandler := NewLogHandlerFunc(*logProviderURL)
+	logHandler := NewLogHandlerFunc(*logProviderURL, time.Minute)
 	testSrv := httptest.NewServer(http.HandlerFunc(logHandler))
 	defer testSrv.Close()
 
@@ -107,7 +108,7 @@ func Test_logsProxyDoesNotLeakGoroutinesWhenClientClosesConnection(t *testing.T)
 
 	logProviderURL, _ := url.Parse(mockLogsUpstreamEndpoint.URL)
 
-	logHandler := NewLogHandlerFunc(*logProviderURL)
+	logHandler := NewLogHandlerFunc(*logProviderURL, time.Minute)
 	testSrv := httptest.NewServer(http.HandlerFunc(logHandler))
 	defer testSrv.Close()