diff --git a/src/index.js b/src/index.js
index b4f295a..9ddd0d4 100644
--- a/src/index.js
+++ b/src/index.js
@@ -59,7 +59,7 @@ export default function parse(md, prevLinks) {
 		}
 		// Code/Indent blocks:
 		else if (token[3] || token[4]) {
-			chunk = '<pre class="code '+(token[4]?'poetry':token[2].toLowerCase())+(token[2] ? ` language-${token[2].toLowerCase()}` : '')+'"><code>'+outdent(encodeAttr(token[3] || token[4]).replace(/^\n+|\n+$/g, ''))+'</code></pre>';
+			chunk = '<pre class="code '+(token[4]?'poetry':token[2].toLowerCase())+'"><code'+(token[2] ? ` class="language-${token[2].toLowerCase()}"` : '')+'>'+outdent(encodeAttr(token[3] || token[4]).replace(/^\n+|\n+$/g, ''))+'</code></pre>';
 		}
 		// > Quotes, -* lists:
 		else if (token[6]) {
diff --git a/test/index.js b/test/index.js
index 3cbb234..1b5ab99 100644
--- a/test/index.js
+++ b/test/index.js
@@ -116,7 +116,7 @@ describe('snarkdown()', () => {
 		it('parses three backtricks (```) as a code block', () => {
 			expect(snarkdown('```\nfunction codeBlocks() {\n\treturn "Can be inserted";\n}\n```')).to.equal('<pre class="code "><code>function codeBlocks() {\n\treturn &quot;Can be inserted&quot;;\n}</code></pre>');
 
-			expect(snarkdown('```js\nfunction codeBlocks() {\n\treturn "Can be inserted";\n}\n```')).to.equal('<pre class="code js language-js"><code>function codeBlocks() {\n\treturn &quot;Can be inserted&quot;;\n}</code></pre>');
+			expect(snarkdown('```js\nfunction codeBlocks() {\n\treturn "Can be inserted";\n}\n```')).to.equal('<pre class="code js"><code class="language-js">function codeBlocks() {\n\treturn &quot;Can be inserted&quot;;\n}</code></pre>');
 		});
 
 		it('parses tabs as a code poetry block', () => {