diff --git a/src/Generators/stubs/feature-test.stub b/src/Generators/stubs/feature-test.stub
index afa4ca5..6d1e990 100644
--- a/src/Generators/stubs/feature-test.stub
+++ b/src/Generators/stubs/feature-test.stub
@@ -1,7 +1,7 @@
 <?php
 namespace {{namespace}};
 
-use TestCase;
+use Tests\TestCase;
 use {{feature_namespace}};
 
 class {{testclass}} extends TestCase
diff --git a/src/Generators/stubs/job-test.stub b/src/Generators/stubs/job-test.stub
index 4b35ed2..1c71cfa 100644
--- a/src/Generators/stubs/job-test.stub
+++ b/src/Generators/stubs/job-test.stub
@@ -2,7 +2,7 @@
 namespace {{namespace}};
 
 use {{job_namespace}};
-use PHPUnit_Framework_TestCase as TestCase;
+use Tests\TestCase;
 
 class {{testclass}} extends TestCase
 {
diff --git a/src/Generators/stubs/operation-test.stub b/src/Generators/stubs/operation-test.stub
index 814fc52..df7090d 100644
--- a/src/Generators/stubs/operation-test.stub
+++ b/src/Generators/stubs/operation-test.stub
@@ -1,7 +1,7 @@
 <?php
 namespace {{namespace}};
 
-use TestCase;
+use Tests\TestCase;
 use {{operation_namespace}};
 
 class {{testclass}} extends TestCase