Code snippets not displaying on search

I have some code snippets to allow learners to copy the code out, but have noticed that sometimes they don't display, or only show some of the text.

I finally realized that the issue is when you open the section through a search. See the gif below - it starts with the code snippet visible, and as soon as you enter search text and hit enter, the code (partially) disappears. Clear the search text and the code is again visible. It doesn't matter if the search text matches with text in the code or elsewhere in the section.

See below, or try it here.