diff --git a/_layouts/stdlib.html b/_layouts/stdlib.html
index 926f83e6..f69fc8dd 100644
--- a/_layouts/stdlib.html
+++ b/_layouts/stdlib.html
@@ -11,11 +11,10 @@ <h3>Sdtlib Diagram Code</h3>
 
 This diagram was created from the following PlantUML code:
 
-<pre>
-    <code class="language-puml">
-{% include_relative input/{{ page.name }}.puml %}
-    </code>
-</pre>
+{% capture my_code %}{% include_relative input/{{ page.name }}.puml %}{% endcapture %}
+{% highlight plantuml %}
+   {{ my_code | strip }}
+{% endhighlight %}
 
 <h3>Available Themes</h3>
 
@@ -39,4 +38,4 @@ <h4 id="{{ skin.name }}">{{ page.display_name }} shown in {{ skin.display_name }
 
 {% include figure.html theme=skin diagram=page url=url %}
 
-{% endfor %}
\ No newline at end of file
+{% endfor %}