Restrict media types by permission for entity reference field widget

I don't know the internals enough to explain why, but one particular line in this patch is causing issues on our side. My wild guess is that maybe in previous versions of Drupal a particular array was associative, but now it's a flat array. Our project also has a ton of patches though, so it's also possible that some other patch is causing this problem. Anyway I'm attaching a version of the patch that seems to get around this problem for us. I changed this line:

$filtered_media_type_ids = array_intersect_key($allowed_media_type_ids, array_flip($permitted_medias));

To this:

$filtered_media_type_ids = array_intersect($allowed_media_type_ids, $permitted_medias);