diff --git a/tests/Pest.php b/tests/Pest.php
index 9103936..30daff3 100644
--- a/tests/Pest.php
+++ b/tests/Pest.php
@@ -1,5 +1,7 @@
 <?php
 
+use ArchTech\REPLACE\Tests\TestCase;
+
 /*
 |--------------------------------------------------------------------------
 | Test Case
@@ -11,7 +13,7 @@
 |
 */
 
-uses(ArchTech\REPLACE\Tests\TestCase::class)->in('Pest');
+uses(TestCase::class)->in('Pest');
 
 /*
 |--------------------------------------------------------------------------
@@ -43,3 +45,8 @@ function something()
 {
     // ..
 }
+
+function pest(): TestCase
+{
+    return Pest\TestSuite::getInstance()->test;
+}