diff --git a/src/Rector/MakeNewlineAfterIfRector.php b/src/Rector/MakeNewlineAfterIfRector.php
index 8dfea09..3ac5321 100644
--- a/src/Rector/MakeNewlineAfterIfRector.php
+++ b/src/Rector/MakeNewlineAfterIfRector.php
@@ -5,7 +5,7 @@
 namespace Rector\RectorCompatTests\Rector;
 
 use PhpParser\Node;
-use PhpParser\Node\Stmt\Class_;
+use PhpParser\Node\Stmt;
 use PhpParser\Node\Stmt\If_;
 use PhpParser\Node\Stmt\Nop;
 use Rector\Contract\PhpParser\Node\StmtsAwareInterface;
@@ -14,15 +14,20 @@
 final class MakeNewlineAfterIfRector extends AbstractRector
 {
     /**
-     * @return array<class-string<Class_>>
+     * @return array<class-string<Node>>
      */
     public function getNodeTypes(): array
     {
-        return [StmtsAwareInterface::class];
+        // local PHPUnit
+        if (is_dir(__DIR__ . '/../../vendor/phpunit')) {
+            return [StmtsAwareInterface::class];
+        }
+
+        return [If_::class];
     }
 
     /**
-     * @param StmtsAwareInterface $node
+     * @param StmtsAwareInterface|If_ $node
      * @return array<Node>|null
      */
     public function refactor(Node $node): ?array