diff --git a/tests/TypeAnnotationTest.php b/tests/TypeAnnotationTest.php
index 4d15c612e05..c55a5911221 100644
--- a/tests/TypeAnnotationTest.php
+++ b/tests/TypeAnnotationTest.php
@@ -595,12 +595,8 @@ class C extends A {}
             ],
             'importedTypeUsedInAssertion' => [
                 'code' => '<?php
-                    /**
-                     * @psalm-type Foo = string
-                     */
-                    class A
-                    {
-                    }
+                    /** @psalm-type Foo = string */
+                    class A {}
 
                     /**
                      * @psalm-immutable