Fix missing end of code block