diff --git a/plugins/code_block.rb b/plugins/code_block.rb
index ff242aa..55267a3 100644
--- a/plugins/code_block.rb
+++ b/plugins/code_block.rb
@@ -64,7 +64,7 @@ module Jekyll
         @file = $1
         @caption = "<figcaption><span>#{$1}</span></figcaption>\n"
       end
-      if @file =~ /\S[\S\s]*\.(\w+)/
+      if @file =~ /\S[\S\s]*\w+\.(\w+)/
         @filetype = $1
       end
       super
@@ -82,7 +82,7 @@ module Jekyll
         @filetype = 'yaml' if @filetype == 'yml'
         source += " #{highlight(code, @filetype)}</figure></div>"
       else
-        source += "<pre><code>" + code.lstrip.rstrip.gsub(/</,'&lt;') + "</code></pre></figure></div>"
+        source += "#{tableize_code(code.lstrip.rstrip.gsub(/</,'&lt;'))}</figure></div>"
       end
       source = source + context['pygments_suffix'] if context['pygments_suffix']
     end
diff --git a/plugins/octopress_filters.rb b/plugins/octopress_filters.rb
index 9c94f0e..1170f8b 100644
--- a/plugins/octopress_filters.rb
+++ b/plugins/octopress_filters.rb
@@ -26,14 +26,17 @@ module OctopressFilters
   #     code snippet
   # ```
   def backtick_codeblock(input)
+    code = nil
     # Markdown support
     input = input.gsub /<p>`{3}\s*(\w+)?<\/p>\s*<pre><code>\s*(.+?)\s*<\/code><\/pre>\s*<p>`{3}<\/p>/m do
       lang = $1
       if lang != ''
         str  = $2.gsub('&lt;','<').gsub('&gt;','>').gsub('&amp;','&')
-        highlight(str, lang)
+        code = highlight(str, lang)
+        "<figure role=code>#{code}</figure>"
       else
-        "<pre><code>#{$2}</code></pre>"
+        code = tableize_code($2)
+        "<figure role=code>#{code}</figure>"
       end
     end
 
@@ -48,9 +51,11 @@ module OctopressFilters
       lang = $1
       str  = $2.gsub(/^\s{4}/, '')
       if lang != ''
-        highlight(str, lang)
+        code = highlight(str, lang)
+        "<figure role=code>#{code}</figure>"
       else
-        "<pre><code>#{$2.gsub('<','&lt;').gsub('>','&gt;')}</code></pre>"
+        code = tableize_code($2.gsub('<','&lt;').gsub('>','&gt;'))
+        "<figure role=code>#{code}</figure>"
       end
     end
   end
diff --git a/plugins/pygments_code.rb b/plugins/pygments_code.rb
index cdd6c3a..1930ec8 100644
--- a/plugins/pygments_code.rb
+++ b/plugins/pygments_code.rb
@@ -8,13 +8,7 @@ FileUtils.mkdir_p(PYGMENTS_CACHE_DIR)
 module HighlightCode
   def highlight(str, lang)
     str = pygments(str, lang).match(/<pre>(.+)<\/pre>/m)[1].to_s.gsub(/ *$/, '') #strip out divs <div class="highlight">
-    table = '<div class="highlight"><table cellpadding="0" cellspacing="0"><tr><td class="gutter"><pre class="line-numbers">'
-    code = ''
-    str.lines.each_with_index do |line,index|
-      table += "<span class='line'>#{index+1}</span>\n"
-      code  += "<div class='line'>#{line}</div>"
-    end
-    table += "</pre></td><td class='code' width='100%'><pre><code class='#{lang}'>#{code}</code></pre></td></tr></table></div>"
+    tableize_code(str, lang)
   end
 
   def pygments(code, lang)
@@ -31,4 +25,13 @@ module HighlightCode
     end
     highlighted_code
   end
+  def tableize_code (str, lang = '')
+    table = '<div class="highlight"><table cellpadding="0" cellspacing="0"><tr><td class="gutter"><pre class="line-numbers">'
+    code = ''
+    str.lines.each_with_index do |line,index|
+      table += "<span class='line'>#{index+1}</span>\n"
+      code  += "<div class='line'>#{line}</div>"
+    end
+    table += "</pre></td><td class='code' width='100%'><pre><code class='#{lang}'>#{code}</code></pre></td></tr></table></div>"
+  end
 end