diff --git a/registry/storage/driver/azure/azure.go b/registry/storage/driver/azure/azure.go
index 70771375a..b06b08764 100644
--- a/registry/storage/driver/azure/azure.go
+++ b/registry/storage/driver/azure/azure.go
@@ -382,8 +382,8 @@ func (d *driver) listBlobs(container, virtPath string) ([]string, error) {
 }
 
 func is404(err error) bool {
-	statusCodeErr, ok := err.(azure.UnexpectedStatusCodeError)
-	return ok && statusCodeErr.Got() == http.StatusNotFound
+	statusCodeErr, ok := err.(azure.AzureStorageServiceError)
+	return ok && statusCodeErr.StatusCode == http.StatusNotFound
 }
 
 type writer struct {