diff --git a/tests/ReturnTypeProvider/ExceptionCodeTest.php b/tests/ReturnTypeProvider/ExceptionCodeTest.php
index f76ca5f2f4a..7b94371dfd3 100644
--- a/tests/ReturnTypeProvider/ExceptionCodeTest.php
+++ b/tests/ReturnTypeProvider/ExceptionCodeTest.php
@@ -39,23 +39,23 @@ function f(\PDOException $e): string {
             'assertions' => [],
         ];
         yield 'CustomThrowable' => [
-            '<?php
+            'code' => '<?php
                 interface CustomThrowable extends \Throwable {}
 
                 /** @var CustomThrowable $e */
                 $code = $e->getCode();
             ',
-            ['$code' => 'int'],
+            'assertions' => ['$code' => 'int'],
         ];
         yield 'Throwable' => [
-            '<?php
+            'code' => '<?php
                 /** @var \Throwable $e */
                 $code = $e->getCode();
             ',
             'assertions' => ['$code' => 'int|string'],
         ];
         yield 'Exception' => [
-            '<?php
+            'code' => '<?php
                 /** @var \Exception $e */
                 $code = $e->getCode();
             ',