Fixing minor code block issue