Fix termination of code block