diff --git a/main.py b/main.py
index 79caa85..5eb9ca5 100644
--- a/main.py
+++ b/main.py
@@ -35,7 +35,7 @@ def md(html, **options):
 
 
 def code_parser(el):
-	return el['data-language'][0] if el.has_attr('data-language') else 'text'
+	return el['data-language'] if el.has_attr('data-language') else 'text'
 
 
 mkdirp('output')