diff --git a/src/stubs/test-policy.stub b/src/stubs/test-policy.stub
index 7b6659b..71c858d 100644
--- a/src/stubs/test-policy.stub
+++ b/src/stubs/test-policy.stub
@@ -4,7 +4,7 @@ namespace Tests\Unit\Policies;
 
 use fullMstr;
 use Illuminate\Foundation\Testing\DatabaseMigrations;
-use Tests\TestCase;
+use Tests\BrowserKitTest as TestCase;
 
 class MasterTest extends TestCase
 {
diff --git a/src/stubs/test-unit.stub b/src/stubs/test-unit.stub
index c501773..de37363 100644
--- a/src/stubs/test-unit.stub
+++ b/src/stubs/test-unit.stub
@@ -4,7 +4,7 @@ namespace Tests\Unit\Models;
 
 use fullMstr;
 use Illuminate\Foundation\Testing\DatabaseMigrations;
-use Tests\TestCase;
+use Tests\BrowserKitTest as TestCase;
 
 class MasterTest extends TestCase
 {
diff --git a/tests/Generators/ModelPolicyTestGeneratorTest.php b/tests/Generators/ModelPolicyTestGeneratorTest.php
index 589fdf3..ca0a40a 100644
--- a/tests/Generators/ModelPolicyTestGeneratorTest.php
+++ b/tests/Generators/ModelPolicyTestGeneratorTest.php
@@ -22,7 +22,7 @@ namespace Tests\Unit\Policies;
 
 use {$this->full_model_name};
 use Illuminate\Foundation\Testing\DatabaseMigrations;
-use Tests\TestCase;
+use Tests\BrowserKitTest as TestCase;
 
 class {$this->model_name}Test extends TestCase
 {
diff --git a/tests/Generators/ModelTestGeneratorTest.php b/tests/Generators/ModelTestGeneratorTest.php
index a5ece64..82b2d21 100644
--- a/tests/Generators/ModelTestGeneratorTest.php
+++ b/tests/Generators/ModelTestGeneratorTest.php
@@ -18,7 +18,7 @@ namespace Tests\Unit\Models;
 
 use {$this->full_model_name};
 use Illuminate\Foundation\Testing\DatabaseMigrations;
-use Tests\TestCase;
+use Tests\BrowserKitTest as TestCase;
 
 class {$this->model_name}Test extends TestCase
 {