diff --git a/bibtexbrowser-test.php b/bibtexbrowser-test.php
index d8cc0d1..4eb9939 100755
--- a/bibtexbrowser-test.php
+++ b/bibtexbrowser-test.php
@@ -759,7 +759,7 @@ class BTBTest extends PHPUnit_Framework_TestCase {
         $d = new SimpleDisplay();
         $d->setDB($db);
         ob_start();
-        $d->display();
+        NoWrapper($d);
         $output = ob_get_clean();
         $res = eval("return ".file_get_contents('reference-output-wp-publications.txt').";");
         $this->assertEquals(strip_tags($res['rendered']), "[wp-publications bib=sample.bib all=1] gives:\n".strip_tags($output)."\n");
diff --git a/reference-output-wp-publications.txt b/reference-output-wp-publications.txt
index 4f8699f..800ebe8 100644
--- a/reference-output-wp-publications.txt
+++ b/reference-output-wp-publications.txt
@@ -5,7 +5,8 @@ array (
 <tr class="bibline"><td class="bibref"><a class="bibanchor" name="2"></a>[2]</td><td class="bibitem"><span itemscope="" itemtype="http://schema.org/ScholarlyArticle"><span class="bibtitle"  itemprop="name">A book</span> (<span class="bibauthor"><span itemprop="author" itemtype="http://schema.org/Person">Jane Doe</span></span>), <span itemprop="datePublished">2001</span>.<span class="Z3988" title="ctx_ver=Z39.88-2004&amp;rft_val_fmt=info%3Aofi%2Ffmt%3Akev%3Amtx%3Abook&amp;rft.btitle=A+book&amp;rft.genre=book&amp;rft.pub=&amp;rfr_id=info%3Asid%2F%3Asample.bib&amp;rft.date=2001&amp;rft.au=Jane+Doe"></span></span> <span class="bibmenu"><a class="biburl" title="doo2001" href="http://bibtexbrowser.com/?wp-publications=doo2001">[bibtex]</a></span></td></tr>
 <tr><td colspan="2" class="sheader">2000</td></tr>
 <tr class="bibline"><td class="bibref"><a class="bibanchor" name="1"></a>[1]</td><td class="bibitem"><span itemscope="" itemtype="http://schema.org/ScholarlyArticle"><span class="bibtitle"  itemprop="name">An article</span> (<span class="bibauthor"><span itemprop="author" itemtype="http://schema.org/Person">Jane Doe</span></span>), <span class="bibbooktitle">In <span itemprop="isPartOf">The Wordpress Journal</span></span>, <span itemprop="datePublished">2000</span>.<span class="Z3988" title="ctx_ver=Z39.88-2004&amp;rft_val_fmt=info%3Aofi%2Ffmt%3Akev%3Amtx%3Ajournal&amp;rft.atitle=An+article&amp;rft.jtitle=The+Wordpress+Journal&amp;rft.volume=&amp;rft.issue=&amp;rft.pub=&amp;rfr_id=info%3Asid%2F%3Asample.bib&amp;rft.date=2000&amp;rft.au=Jane+Doe"></span></span> <span class="bibmenu"><a class="biburl" title="doe2000" href="http://bibtexbrowser.com/?wp-publications=doe2000">[bibtex]</a></span></td></tr>
-</table>    <script type="text/x-mathjax-config">
+</table>
+    <script type="text/x-mathjax-config">
       MathJax.Hub.Config({
         tex2jax: {inlineMath: [["$","$"]]}
       });