diff --git a/src/Tests/ViewFieldUITest.php b/src/Tests/ViewFieldUITest.php
index 40deb29..bc0dfbe 100644
--- a/src/Tests/ViewFieldUITest.php
+++ b/src/Tests/ViewFieldUITest.php
@@ -3,7 +3,7 @@
 namespace Drupal\views_field_view\Tests;
 
 use Drupal\views\Tests\ViewTestData;
-use Drupal\views_ui\Tests\UITestBase;
+use Drupal\Tests\views_ui\Functional\UITestBase;
 
 /**
  * Tests the UI of views_field_view.
@@ -38,8 +38,8 @@ class ViewFieldUITest extends UITestBase {
   /**
    * {@inheritdoc}
    */
-  protected function setUp() {
-    parent::setUp();
+  protected function setUp($import_test_views = TRUE) {
+    parent::setUp($import_test_views = TRUE);
 
     ViewTestData::createTestViews(get_class($this), ['views_field_view_test_config']);
   }