Bug fix for code block