Fix small code blocks issue