diff --git a/tests/Unit/BuilderTest.php b/tests/Unit/BuilderTest.php
index d8a83d0..eb66502 100644
--- a/tests/Unit/BuilderTest.php
+++ b/tests/Unit/BuilderTest.php
@@ -2,8 +2,9 @@
 
 use Opdavies\GmailFilterBuilder\Builder;
 use Opdavies\GmailFilterBuilder\Filter;
+use PHPUnit\Framework\TestCase;
 
-class BuilderTest extends \PHPUnit_Framework_TestCase
+class BuilderTest extends TestCase
 {
     public function testBuild()
     {
diff --git a/tests/Unit/FilterTest.php b/tests/Unit/FilterTest.php
index 1dd3a1c..b99aa7b 100644
--- a/tests/Unit/FilterTest.php
+++ b/tests/Unit/FilterTest.php
@@ -3,8 +3,9 @@
 namespace Opdavies\Tests\GmailFilterBuilder;
 
 use Opdavies\GmailFilterBuilder\Filter;
+use PHPUnit\Framework\TestCase;
 
-class FilterTest extends \PHPUnit_Framework_TestCase
+class FilterTest extends TestCase
 {
     /**
      * @var Filter