diff --git a/src/Generators/TestGenerator.php b/src/Generators/TestGenerator.php
index b6648952..b8ce0a5b 100644
--- a/src/Generators/TestGenerator.php
+++ b/src/Generators/TestGenerator.php
@@ -463,8 +463,8 @@ private function addFakerTrait(Controller $controller)
 
     private function addTestAssertionsTrait(Controller $controller)
     {
-        $this->addImport($controller, 'JMac\\Testing\\Traits\HttpTestAssertions');
-        $this->addTrait($controller, 'HttpTestAssertions');
+        $this->addImport($controller, 'JMac\\Testing\\Traits\AdditionalAssertions');
+        $this->addTrait($controller, 'AdditionalAssertions');
     }
 
     private function addRefreshDatabaseTrait(Controller $controller)
diff --git a/tests/fixtures/tests/readme-example.php b/tests/fixtures/tests/readme-example.php
index 6187a715..c5804de9 100644
--- a/tests/fixtures/tests/readme-example.php
+++ b/tests/fixtures/tests/readme-example.php
@@ -11,7 +11,7 @@
 use Illuminate\Support\Facades\Event;
 use Illuminate\Support\Facades\Mail;
 use Illuminate\Support\Facades\Queue;
-use JMac\Testing\Traits\HttpTestAssertions;
+use JMac\Testing\Traits\AdditionalAssertions;
 use Tests\TestCase;
 
 /**
@@ -19,7 +19,7 @@
  */
 class PostControllerTest extends TestCase
 {
-    use HttpTestAssertions, RefreshDatabase, WithFaker;
+    use AdditionalAssertions, RefreshDatabase, WithFaker;
 
     /**
      * @test
diff --git a/tests/fixtures/tests/reference-cache.php b/tests/fixtures/tests/reference-cache.php
index e07021bd..7e3bddca 100644
--- a/tests/fixtures/tests/reference-cache.php
+++ b/tests/fixtures/tests/reference-cache.php
@@ -5,7 +5,7 @@
 use App\User;
 use Illuminate\Foundation\Testing\RefreshDatabase;
 use Illuminate\Foundation\Testing\WithFaker;
-use JMac\Testing\Traits\HttpTestAssertions;
+use JMac\Testing\Traits\AdditionalAssertions;
 use Tests\TestCase;
 
 /**
@@ -13,7 +13,7 @@
  */
 class UserControllerTest extends TestCase
 {
-    use HttpTestAssertions, RefreshDatabase, WithFaker;
+    use AdditionalAssertions, RefreshDatabase, WithFaker;
 
     /**
      * @test