Markdown preview - how to wrap text in a code block (or atom-text-editor)

Hi,

I want the markdown preview wrap text in code blocks, basically revert this change.

I thought it would be a simple job but then asking this question made me realise that the way the code blocks are styled is completely different now and a custom tag atom-text-editor is used for code blocks.

How can I enable wrapping in that editor?