diff --git a/tests/Unit/BuilderTest.php b/tests/Unit/BuilderTest.php
index cf70229..b7ccea5 100644
--- a/tests/Unit/BuilderTest.php
+++ b/tests/Unit/BuilderTest.php
@@ -1,7 +1,7 @@
 <?php
 
-use Opdavies\GmailFilterBuilder\Builder;
-use Opdavies\GmailFilterBuilder\Filter;
+use Opdavies\GmailFilterBuilder\Model\Filter;
+use Opdavies\GmailFilterBuilder\Service\Builder;
 use PHPUnit\Framework\TestCase;
 
 class BuilderTest extends TestCase
diff --git a/tests/Unit/FilterTest.php b/tests/Unit/FilterTest.php
index f986900..aaded0c 100644
--- a/tests/Unit/FilterTest.php
+++ b/tests/Unit/FilterTest.php
@@ -2,7 +2,7 @@
 
 namespace Opdavies\Tests\GmailFilterBuilder;
 
-use Opdavies\GmailFilterBuilder\Filter;
+use Opdavies\GmailFilterBuilder\Model\Filter;
 use PHPUnit\Framework\TestCase;
 
 class FilterTest extends TestCase