diff --git a/plugins/code_block.rb b/plugins/code_block.rb
index af64e03..cdb1230 100644
--- a/plugins/code_block.rb
+++ b/plugins/code_block.rb
@@ -75,6 +75,7 @@ module Jekyll
       code = super.join
       source = "<div><figure role=code>"
       source += @caption if @caption
+      source = context['pygments_prefix'] + source if context['pygments_prefix']
       if @filetype
         @filetype = 'objc' if @filetype == 'm'
         @filetype = 'perl' if @filetype == 'pl'
@@ -82,6 +83,7 @@ module Jekyll
       else
         source += "<pre><code>" + code.lstrip.rstrip.gsub(/</,'&lt;') + "</code></pre></figure></div>"
       end
+      source = source + context['pygments_suffix'] if context['pygments_suffix']
       partial = Liquid::Template.parse(source)
       context.stack do
         partial.render(context)