diff --git a/plugins/code_block.rb b/plugins/code_block.rb
index 00b0b43..bf89fea 100644
--- a/plugins/code_block.rb
+++ b/plugins/code_block.rb
@@ -24,7 +24,7 @@
 #
 # Output:
 #
-# <figure role=code>
+# <figure class='code'>
 # <figcaption><span>Got pain? painrelief.sh</span> <a href="http://site.com/painrelief.sh">Download it!</a>
 # <div class="highlight"><pre><code class="sh">
 # -- nicely escaped highlighted code --
@@ -37,7 +37,7 @@
 # <sarcasm>Ooooh, sarcasm... How original!</sarcasm>
 # {% endcodeblock %}
 #
-# <figure role=code>
+# <figure class='code'>
 # <pre><code>&lt;sarcasm> Ooooh, sarcasm... How original!&lt;/sarcasm></code></pre>
 # </figure>
 #
@@ -80,7 +80,7 @@ module Jekyll
     def render(context)
       output = super
       code = super.join
-      source = "<figure role=code>"
+      source = "<figure class='code'>"
       source += @caption if @caption
       if @filetype
         source += " #{highlight(code, @filetype)}</figure>"
diff --git a/plugins/include_code.rb b/plugins/include_code.rb
index 80951cb..fc6daa3 100644
--- a/plugins/include_code.rb
+++ b/plugins/include_code.rb
@@ -61,7 +61,7 @@ module Jekyll
         @filetype = file.extname.sub('.','') if @filetype.nil?
         title = @title ? "#{@title} (#{file.basename})" : file.basename
         url = "/#{code_dir}/#{@file}"
-        source = "<figure role=code><figcaption><span>#{title}</span> <a href='#{url}'>download</a></figcaption>\n"
+        source = "<figure class='code'><figcaption><span>#{title}</span> <a href='#{url}'>download</a></figcaption>\n"
         source += " #{highlight(code, @filetype)}</figure>"
         safe_wrap(source)
       end