Fix missing lines for code block