Ordered lists with highlight shorcode

Is there a way to continue the numbering in an ordered list with a code block between items using the highlight short code?

Wrapping the shortcode in a div works.

A post was split to a new topic: Put a blockquote into a numbered list