diff --git a/src/Tests/FlagTestBase.php b/src/Tests/FlagTestBase.php
index 8433cf7..45c62be 100644
--- a/src/Tests/FlagTestBase.php
+++ b/src/Tests/FlagTestBase.php
@@ -7,7 +7,7 @@ use Drupal\simpletest\WebTestBase;
 use Drupal\flag\Entity\Flag;
 use Drupal\Core\Template\Attribute;
 use Drupal\flag\Tests\FlagWithFormTrait;
-use Drupal\Tests\flag\Traits\FlagPermissionTrait;
+use Drupal\Tests\flag\Traits\FlagPermissionsTrait;
 
 /**
  * Provides common methods for Flag tests.