diff --git a/packages/phpunit/src/Rector/Class_/ArrayArgumentInTestToDataProviderRector.php b/packages/phpunit/src/Rector/Class_/ArrayArgumentInTestToDataProviderRector.php
index f5b451b2c9c..7e7e084d2fb 100644
--- a/packages/phpunit/src/Rector/Class_/ArrayArgumentInTestToDataProviderRector.php
+++ b/packages/phpunit/src/Rector/Class_/ArrayArgumentInTestToDataProviderRector.php
@@ -21,6 +21,7 @@ use PHPStan\Type\MixedType;
 use PHPStan\Type\Type;
 use PHPStan\Type\UnionType;
 use Rector\AttributeAwarePhpDoc\Ast\PhpDoc\AttributeAwareParamTagValueNode;
+use Rector\AttributeAwarePhpDoc\Ast\PhpDoc\AttributeAwarePhpDocTagNode;
 use Rector\BetterPhpDocParser\PhpDocInfo\PhpDocInfo;
 use Rector\Core\Exception\ShouldNotHappenException;
 use Rector\Core\Rector\AbstractPHPUnitRector;
@@ -263,7 +264,9 @@ PHP
 
     private function createDataProviderTagNode(string $dataProviderMethodName): PhpDocTagNode
     {
-        return new PhpDocTagNode('@dataProvider', new GenericTagValueNode($dataProviderMethodName . '()'));
+        return new AttributeAwarePhpDocTagNode('@dataProvider', new GenericTagValueNode(
+            $dataProviderMethodName . '()'
+        ));
     }
 
     /**