diff --git a/import_test.go b/import_test.go
index df7ed24cc..6f5d9b6db 100644
--- a/import_test.go
+++ b/import_test.go
@@ -278,7 +278,8 @@ func checkImages(t *testing.T, target digest.Digest, actual []images.Image, name
 			t.Fatalf("image(%d) unexpected name %q, expected %q", i, actual[i].Name, n)
 		}
 
-		if actual[i].Target.MediaType != ocispec.MediaTypeImageManifest {
+		if actual[i].Target.MediaType != ocispec.MediaTypeImageManifest &&
+			actual[i].Target.MediaType != images.MediaTypeDockerSchema2Manifest {
 			t.Fatalf("image(%d) unexpected media type: %s", i, actual[i].Target.MediaType)
 		}
 	}