diff --git a/composer.json b/composer.json
index 1c5135f..e01a0f0 100755
--- a/composer.json
+++ b/composer.json
@@ -33,7 +33,8 @@
     },
     "autoload": {
         "psr-4": {
-            "Psalm\\PhpUnitPlugin\\": "src"
+            "Psalm\\PhpUnitPlugin\\": "src",
+            "PHPUnit\\Framework\\": "stubs"
         }
     },
     "autoload-dev": {