Fixed a bug with code blocks