Fix wrapping for code blocks