1 files changed, 5 insertions, 1 deletions
diff --git a/mock.go b/mock.go
index 44b8837..147ca82 100644
--- a/mock.go
+++ b/mock.go
@@ -1,3 +1,7 @@
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
 package flyscrape
 
 import (
@@ -7,7 +11,7 @@ import (
 	"strings"
 )
 
-func MockTransport(statusCode int, html string) func(*http.Request) (*http.Response, error) {
+func MockTransport(statusCode int, html string) RoundTripFunc {
 	return func(*http.Request) (*http.Response, error) {
 		return MockResponse(statusCode, html)
 	}