Fix code block styling inside of a preview