diff --git a/tests/src/FunctionalJavascript/WidgetJSTest.php b/tests/src/FunctionalJavascript/WidgetJSTest.php
index b65a019..1f9a767 100644
--- a/tests/src/FunctionalJavascript/WidgetJSTest.php
+++ b/tests/src/FunctionalJavascript/WidgetJSTest.php
@@ -19,7 +19,6 @@ class WidgetJSTest extends JavascriptTestBase {
   public static $modules = [
     'views',
     'search_api',
-    'search_api_test_backend',
     'facets',
     'facets_search_api_dependency',
     'block',
@@ -75,24 +74,25 @@ class WidgetJSTest extends JavascriptTestBase {
 
     // Make sure the block is shown on the page.
     $page = $this->getSession()->getPage();
-    $page_content = $page->getContent();
-    $this->assertTrue(strpos($page_content, 'Llama block') > 0, 'Block found on page.');
+    $block = $page->findById('block-llama-block');
+    $block->isVisible();
 
     // Narrow the context to the block that shows the facet and get objects that
     // contain the <li>-html elements.
-    $list_items = $page->findById('block-llama-block')->findAll('css', 'li');
+    $list_items = $block->findAll('css', 'li');
 
     $this->assertCount(2, $list_items);
 
     /** @var \Behat\Mink\Element\NodeElement $list_item */
     foreach ($list_items as $list_item) {
       $this->assertEquals('li', $list_item->getTagName());
-      $this->assertTrue(strpos($list_item->getHtml(), '<a href') !== FALSE, 'Link found in list item.');
-      $this->assertTrue(strpos($list_item->getHtml(), '<input type="checkbox" class="facets-checkbox"') !== FALSE, 'Link found in list item.');
+      $list_item->find('css', 'a')->isVisible();
+      $list_item->find('css', 'input[type="checkbox"]')->isVisible();
     }
 
     // Check that clicking one of the checkboxes changes the url.
-    $page->findField('item (3)')->click();
+    $checkbox = $page->findField('item (3)');
+    $checkbox->click();
     $current_url = $this->getSession()->getCurrentUrl();
     $this->assertTrue(strpos($current_url, 'search-api-test-fulltext?f%5B0%5D=llama%253Aitem') !== FALSE);
   }