Fix code block issue