diff --git a/distribution/pull_v2_test.go b/distribution/pull_v2_test.go
index 74d9cff0a9..60d8835944 100644
--- a/distribution/pull_v2_test.go
+++ b/distribution/pull_v2_test.go
@@ -331,7 +331,7 @@ func testNewPuller(t *testing.T, rawurl string) *puller {
 	endpoint := registry.APIEndpoint{
 		Mirror:       false,
 		URL:          uri,
-		Version:      2,
+		Version:      registry.APIVersion2,
 		Official:     false,
 		TrimHostname: false,
 		TLSConfig:    nil,
diff --git a/distribution/registry_unit_test.go b/distribution/registry_unit_test.go
index f48418baf2..de56ddd921 100644
--- a/distribution/registry_unit_test.go
+++ b/distribution/registry_unit_test.go
@@ -43,7 +43,7 @@ func testTokenPassThru(t *testing.T, ts *httptest.Server) {
 	endpoint := registrypkg.APIEndpoint{
 		Mirror:       false,
 		URL:          uri,
-		Version:      2,
+		Version:      registrypkg.APIVersion2,
 		Official:     false,
 		TrimHostname: false,
 		TLSConfig:    nil,