Colab virtual screen in Reinforcement Learning Course

Iโ€™m running the colab tutorial for this course Let's train our first Deep Reinforcement Learning agent and upload it to the Hub ๐Ÿš€ - Hugging Face Course. But I found the virtual screen is not working properly in the Colab. I think itโ€™s expected to display something like actual screen, Instead of something like the one below: