diff --git a/src/Generators/PestTestGenerator.php b/src/Generators/PestTestGenerator.php
index d9a8e93a..57c3e35f 100644
--- a/src/Generators/PestTestGenerator.php
+++ b/src/Generators/PestTestGenerator.php
@@ -4,6 +4,7 @@
 
 use Blueprint\Blueprint;
 use Blueprint\Concerns\HandlesImports;
+use Blueprint\Concerns\HandlesTraits;
 use Blueprint\Contracts\Generator;
 use Blueprint\Contracts\Model as BlueprintModel;
 use Blueprint\Models\Column;
@@ -25,7 +26,7 @@
 
 class PestTestGenerator extends AbstractClassGenerator implements Generator
 {
-    use HandlesImports;
+    use HandlesImports, HandlesTraits;
 
     const TESTS_VIEW = 1;
 
@@ -41,8 +42,6 @@ class PestTestGenerator extends AbstractClassGenerator implements Generator
 
     protected array $types = ['controllers', 'tests'];
 
-    protected array $traits = [];
-
     public function output(Tree $tree): array
     {
         $this->tree = $tree;
@@ -658,11 +657,6 @@ private function buildLines($lines): string
         return str_pad(' ', 4) . implode(PHP_EOL . str_pad(' ', 4), $lines);
     }
 
-    protected function addTrait(BlueprintModel $model, $trait): void
-    {
-        $this->traits[$model->name()][] = $trait;
-    }
-
     protected function buildTraits(BlueprintModel $model): string
     {
         if (empty($this->traits[$model->name()])) {