diff --git a/tests/EnumTest.php b/tests/EnumTest.php
index f1b74ba4ba9..963614bd198 100644
--- a/tests/EnumTest.php
+++ b/tests/EnumTest.php
@@ -419,14 +419,17 @@ function foo(): Code|null
                     }
 
                     $code = foo();
+                    $code1 = null;
+                    $code2 = null;
                     if($code instanceof Code){
                         $code1 = $code;
-                    } else {
+                    }
+                    if(!$code instanceof Code){
                         $code2 = $code;
                     }',
                 'assertions' => [
-                    '$code1' => 'Code|Code::Ok',
-                    '$code2' => 'never',
+                    '$code1' => 'Code|null',
+                    '$code2' => 'null',
                 ],
                 'ignored_issues' => [],
                 'php_version' => '8.1',