diff --git a/client/gutenberg/editor/hooks/components/media-upload/index.js b/client/gutenberg/editor/hooks/components/media-upload/index.js
index <HASH>..<HASH> 100644
--- a/client/gutenberg/editor/hooks/components/media-upload/index.js
+++ b/client/gutenberg/editor/hooks/components/media-upload/index.js
@@ -77,19 +77,17 @@ export class MediaUpload extends Component {
 	};
 
 	getEnabledFilters = () => {
-		// TODO: Replace with `allowedTypes` (array) after updating Gutenberg
-		const { type } = this.props;
-		switch ( type ) {
-			case 'image':
-				return [ 'images' ];
-			case 'audio':
-				return [ 'audio' ];
-			case 'video':
-				return [ 'videos' ];
-			case 'document':
-				return [ 'documents' ];
-		}
-		return undefined;
+		const { allowedTypes } = this.props;
+
+		const enabledFiltersMap = {
+			image: 'images',
+			audio: 'audio',
+			video: 'videos',
+		};
+
+		return isArray( allowedTypes )
+			? allowedTypes.map( type => enabledFiltersMap[ type ] )
+			: undefined;
 	};
 
 	getSelectedItems = () => {
Gutenberg: wire up Media library enabled filters (#<I>)

The existing mapping of allowed types to enabled filters was no longer

accurate after the package update. This fixes the mapping and refactors

the code to handle the cases when multiple allowed types can be returned.