diff --git a/app/views/style_guide/style/_partial.erb b/app/views/style_guide/style/_partial.erb
index f29168c..ce3641b 100644
--- a/app/views/style_guide/style/_partial.erb
+++ b/app/views/style_guide/style/_partial.erb
@@ -23,7 +23,7 @@
 
   <input id="style-guide-toggle-<%= partial.id %>" type="checkbox" />
   <label for="style-guide-toggle-<%= partial.id %>">
-    <pre class="style-guide-partial-source prettyprint linenums"><%= escape_for_display(partial.render) %></pre>
+    <pre class="style-guide-partial-source prettyprint linenums"><%= escape_for_display(partial.render_source_code) %></pre>
     <div class="style-guide-toggle show">Show source code</div>
     <div class="style-guide-toggle hide">Hide source code</div>
   </label>
diff --git a/lib/style_guide/partial.rb b/lib/style_guide/partial.rb
index d4a9c33..b700513 100644
--- a/lib/style_guide/partial.rb
+++ b/lib/style_guide/partial.rb
@@ -43,7 +43,11 @@ def identifiers
     end
 
     def render
-      @render ||= action_view.render(:file => path)
+      @render = action_view.render(file: path)
+    end
+
+    def render_source_code
+      @render = Nokogiri::HTML.fragment(render, 'utf-8').to_xhtml
     end
 
     private
diff --git a/spec/lib/style_guide/partial_spec.rb b/spec/lib/style_guide/partial_spec.rb
index a612917..029293c 100644
--- a/spec/lib/style_guide/partial_spec.rb
+++ b/spec/lib/style_guide/partial_spec.rb
@@ -117,4 +117,14 @@
 
     it { should == "hi" }
   end
+
+  describe "#render_source" do
+    let(:mock_view) { double(:view, :render => '<h1>Hello world</h1><div><h2>More stuff</h2><div><p>Mooooore stuff</p></div></div>') }
+
+    before { partial.stub(:action_view).and_return(mock_view) }
+
+    subject { partial.render_source_code }
+
+    it { should == "<h1>Hello world</h1><div>\n  <h2>More stuff</h2>\n  <div>\n    <p>Mooooore stuff</p>\n  </div>\n</div>" }
+  end
 end