X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/63cb6015a8c6db453f5551e50c04bf6ebbc0f3f1..refs/pull/3593/head:/tests/Entity/PageRevisionTest.php

diff --git a/tests/Entity/PageRevisionTest.php b/tests/Entity/PageRevisionTest.php
index ce203ea36..cc4900249 100644
--- a/tests/Entity/PageRevisionTest.php
+++ b/tests/Entity/PageRevisionTest.php
@@ -211,11 +211,11 @@ class PageRevisionTest extends TestCase
         $this->asAdmin()->put($page->getUrl(), ['name' => 'Updated page', 'html' => 'new page html']);
 
         $resp = $this->get($page->refresh()->getUrl('/revisions'));
-        $resp->assertElementContains('td', '(WYSIWYG)');
-        $resp->assertElementNotContains('td', '(Markdown)');
+        $this->withHtml($resp)->assertElementContains('td', '(WYSIWYG)');
+        $this->withHtml($resp)->assertElementNotContains('td', '(Markdown)');
 
         $this->asAdmin()->put($page->getUrl(), ['name' => 'Updated page', 'markdown' => '# Some markdown content']);
         $resp = $this->get($page->refresh()->getUrl('/revisions'));
-        $resp->assertElementContains('td', '(Markdown)');
+        $this->withHtml($resp)->assertElementContains('td', '(Markdown)');
     }
 }