diff --git a/pkg/skaffold/docker/context_test.go b/pkg/skaffold/docker/context_test.go
index 5b69d76ddca..7699f19b6f8 100644
--- a/pkg/skaffold/docker/context_test.go
+++ b/pkg/skaffold/docker/context_test.go
@@ -31,6 +31,11 @@ func TestDockerContext(t *testing.T) {
 	tmpDir, cleanup := testutil.TempDir(t)
 	defer cleanup()
 
+	RetrieveImage = mockRetrieveImage
+	defer func() {
+		RetrieveImage = retrieveImage
+	}()
+
 	os.Mkdir(filepath.Join(tmpDir, "files"), 0750)
 	ioutil.WriteFile(filepath.Join(tmpDir, "files", "ignored.txt"), []byte(""), 0644)
 	ioutil.WriteFile(filepath.Join(tmpDir, "files", "included.txt"), []byte(""), 0644)