diff --git a/tests/src/FunctionalJavascript/MaxlengthCustomWidgetTest.php b/tests/src/FunctionalJavascript/MaxlengthCustomWidgetTest.php
index d3e4576..9676c94 100644
--- a/tests/src/FunctionalJavascript/MaxlengthCustomWidgetTest.php
+++ b/tests/src/FunctionalJavascript/MaxlengthCustomWidgetTest.php
@@ -3,14 +3,14 @@
 namespace Drupal\Tests\maxlength\FunctionalJavascript;
 
 use Drupal\Core\Entity\Entity\EntityFormDisplay;
-use Drupal\FunctionalJavascriptTests\JavascriptTestBase;
+use Drupal\FunctionalJavascriptTests\WebDriverTestBase;
 
 /**
  * Tests the custom widget support.
  *
  * @group maxlength
  */
-class MaxlengthCustomWidgetTest extends JavascriptTestBase {
+class MaxlengthCustomWidgetTest extends WebDriverTestBase {
 
   /**
    * {@inheritdoc}
diff --git a/tests/src/FunctionalJavascript/MaxlengthJavascriptTest.php b/tests/src/FunctionalJavascript/MaxlengthJavascriptTest.php
index 04caf2b..e3173ba 100644
--- a/tests/src/FunctionalJavascript/MaxlengthJavascriptTest.php
+++ b/tests/src/FunctionalJavascript/MaxlengthJavascriptTest.php
@@ -6,14 +6,14 @@ use Drupal\Core\Entity\Entity\EntityFormDisplay;
 use Drupal\entity_test\Entity\EntityTest;
 use Drupal\field\Entity\FieldConfig;
 use Drupal\field\Entity\FieldStorageConfig;
-use Drupal\FunctionalJavascriptTests\JavascriptTestBase;
+use Drupal\FunctionalJavascriptTests\WebDriverTestBase;
 
 /**
  * Tests Javascript behaviour of Maxlength module.
  *
  * @group maxlength
  */
-class MaxlengthJavascriptTest extends JavascriptTestBase {
+class MaxlengthJavascriptTest extends WebDriverTestBase {
 
   /**
    * {@inheritdoc}