Fix tests for code blocks