Fixed issue with code block