Minor fix for codeblock