I've just launched a plugin that will add this functionality back in, along with a number of other recently removed WordPress functions. Each of them is switchable, so you can choose which features you'd like.

https://wordpress.org/plugins/restore-lost-functionality/

David.