Media browser: selected items lose "checked" class after submission

In media browser the selected items loses the checked class after submission, ex:

We have media field with only one element allowed to be selected, if the user selected 2 elements then submitted the media browser form it will display an error "Only 1 item allowed" and the selected items will stay selected without having the checked class.

User will not be able to know which items are still selected to reset them unless he reloaded the page.