There's no reason it can't be added back. It's one chunk of code. I'll revert it myself if it's that difficult (it's not).