Fix display_code_blocks