Code blocks fix