diff --git a/rules-tests/CodeQuality/Rector/Class_/YieldDataProviderRector/Fixture/use_data_provider_with_phpdoc.php.inc b/rules-tests/CodeQuality/Rector/Class_/YieldDataProviderRector/Fixture/use_data_provider_with_phpdoc.php.inc
index 77decd6f..e98c3430 100644
--- a/rules-tests/CodeQuality/Rector/Class_/YieldDataProviderRector/Fixture/use_data_provider_with_phpdoc.php.inc
+++ b/rules-tests/CodeQuality/Rector/Class_/YieldDataProviderRector/Fixture/use_data_provider_with_phpdoc.php.inc
@@ -20,6 +20,7 @@ final class UseDataProviderTestPhpdoc extends TestCase
         ];
     }
 
+    /** @return list<array<string>> */
     public static function dataProvider()
     {
         return [
@@ -51,6 +52,7 @@ final class UseDataProviderTestPhpdoc extends TestCase
         yield ['<?php implode("", $foo, );', '<?php implode($foo, "", );'];
     }
 
+    /** @return \Iterator<int<0, max>, array<string>> */
     public static function dataProvider(): \Iterator
     {
         yield ['<?php implode(\'\', $foo, );', '<?php implode($foo, );'];
diff --git a/rules/CodeQuality/Rector/Class_/YieldDataProviderRector.php b/rules/CodeQuality/Rector/Class_/YieldDataProviderRector.php
index b5fc84ef..bc0273d8 100644
--- a/rules/CodeQuality/Rector/Class_/YieldDataProviderRector.php
+++ b/rules/CodeQuality/Rector/Class_/YieldDataProviderRector.php
@@ -11,7 +11,6 @@
 use PhpParser\Node\Stmt\ClassMethod;
 use PhpParser\Node\Stmt\Return_;
 use PHPStan\PhpDocParser\Ast\PhpDoc\ReturnTagValueNode;
-use PHPStan\Type\ArrayType;
 use PHPStan\Type\Generic\GenericObjectType;
 use Rector\BetterPhpDocParser\PhpDocInfo\PhpDocInfoFactory;
 use Rector\BetterPhpDocParser\PhpDocManipulator\PhpDocTypeChanger;
@@ -177,7 +176,7 @@ private function removeReturnTag(ClassMethod $classMethod): void
             return;
         }
 
-        if ($phpDocInfo->getReturnType() instanceof ArrayType) {
+        if ($phpDocInfo->getReturnType()->isArray()->yes()) {
             $keyType = $phpDocInfo->getReturnType()
                 ->getIterableKeyType();
             $itemType = $phpDocInfo->getReturnType()