The Colab team has fixed it and released a patched runtime!
Everything is working nicely now without any manual hacks: