diff --git a/source/js/main.js b/source/js/main.js
index 7e0493a..65b1e29 100644
--- a/source/js/main.js
+++ b/source/js/main.js
@@ -278,7 +278,7 @@ $(function () {
       var langNameIndex, langName
       $figureHighlight.each(function () {
         langNameIndex = langName = $(this).attr('class').split(' ')[1]
-        if (langNameIndex === 'plain') langName = 'Code'
+        if (langNameIndex === 'plain' || langNameIndex === undefined) langName = 'Code'
         $(this).find('.highlight-tools').append('<div class="code-lang">' + langName + '</div>')
       })
     }