diff --git a/packages/block-library/src/gallery/use-get-media.js b/packages/block-library/src/gallery/use-get-media.js index 2d70119e4aa22..0114db392bc34 100644 --- a/packages/block-library/src/gallery/use-get-media.js +++ b/packages/block-library/src/gallery/use-get-media.js @@ -29,23 +29,18 @@ export default function useGetMedia( innerBlockImages ) { if ( imageIds.length === 0 ) { return currentImageMedia; } - const getMedia = select( coreStore ).getMedia; - const newImageMedia = imageIds.map( ( img ) => { - return getMedia( img ); - } ); - if ( newImageMedia.some( ( img ) => ! img ) ) { - return currentImageMedia; - } - - return newImageMedia; + return select( coreStore ).getMediaItems( { + include: imageIds.join( ',' ), + per_page: -1, + } ); }, [ innerBlockImages ] ); if ( - imageMedia?.length !== currentImageMedia.length || - imageMedia.some( + imageMedia?.length !== currentImageMedia?.length || + imageMedia?.some( ( newImage ) => ! currentImageMedia.find( ( currentImage ) => currentImage.id === newImage.id