diff --git a/src/Expectation.php b/src/Expectation.php
index 1bc20abd..22f79122 100644
--- a/src/Expectation.php
+++ b/src/Expectation.php
@@ -4,6 +4,7 @@ declare(strict_types=1);
 
 namespace Pest;
 
+use Attribute;
 use BadMethodCallException;
 use Closure;
 use InvalidArgumentException;
@@ -837,7 +838,7 @@ final class Expectation
     /**
      * Asserts that the given expectation target to have the given attribute.
      *
-     * @param  class-string  $attribute
+     * @param  class-string<Attribute>  $attribute
      */
     public function toHaveAttribute(string $attribute): ArchExpectation
     {
diff --git a/src/Expectations/OppositeExpectation.php b/src/Expectations/OppositeExpectation.php
index 4953c1af..95aae090 100644
--- a/src/Expectations/OppositeExpectation.php
+++ b/src/Expectations/OppositeExpectation.php
@@ -4,6 +4,7 @@ declare(strict_types=1);
 
 namespace Pest\Expectations;
 
+use Attribute;
 use Pest\Arch\Contracts\ArchExpectation;
 use Pest\Arch\Expectations\Targeted;
 use Pest\Arch\Expectations\ToBeUsedIn;
@@ -381,7 +382,7 @@ final class OppositeExpectation
     /**
      * Asserts that the given expectation target not to have the given attribute.
      *
-     * @param  class-string  $attribute
+     * @param  class-string<Attribute>  $attribute
      */
     public function toHaveAttribute(string $attribute): ArchExpectation
     {