diff --git a/src/Tests/FlagCountsTest.php b/src/Tests/FlagCountsTest.php
index 0a7b5c4..5bcedaa 100644
--- a/src/Tests/FlagCountsTest.php
+++ b/src/Tests/FlagCountsTest.php
@@ -6,10 +6,8 @@
 
 namespace Drupal\flag\Tests;
 
-use Drupal\simpletest\WebTestBase;
 use Drupal\flag\Tests\FlagTestBase;
 use Drupal\node\Entity\Node;
-use Drupal\flag\Entity\Flag;
 
 /**
  * Tests the Flag counts API.