diff --git a/composer.json b/composer.json
index c024582..e1dde98 100755
--- a/composer.json
+++ b/composer.json
@@ -27,9 +27,7 @@
     },
     "autoload": {
         "psr-4": {
-            "Psalm\\PhpUnitPlugin\\": ["."],
-            "Psalm\\PhpUnitPlugin\\Hooks\\": ["hooks"],
-            "Psalm\\PhpUnitPlugin\\Exception\\" : ["Exception"]
+            "Psalm\\PhpUnitPlugin\\": "src"
         }
     },
     "autoload-dev": {
diff --git a/phpcs.xml.dist b/phpcs.xml.dist
index 729276b..ac7580c 100755
--- a/phpcs.xml.dist
+++ b/phpcs.xml.dist
@@ -5,11 +5,9 @@
     <arg name="colors"/>
 
     <!-- Paths to check -->
-    <file>Plugin.php</file>
+    <file>src</file>
     <file>tests/_support/Helper</file>
     <file>tests/_support/AcceptanceTester.php</file>
-    <file>hooks</file>
-    <file>Exception</file>
 
 
     <!-- inherit rules from: -->
diff --git a/psalm.xml.dist b/psalm.xml.dist
index 633d409..44f67aa 100755
--- a/psalm.xml.dist
+++ b/psalm.xml.dist
@@ -6,12 +6,7 @@
     xsi:schemaLocation="https://getpsalm.org/schema/config vendor/vimeo/psalm/config.xsd"
 >
     <projectFiles>
-        <directory name="." />
-        <ignoreFiles>
-            <directory name="vendor" />
-            <directory name="stubs" />
-            <directory name="tests/_support" />
-        </ignoreFiles>
+        <directory name="src" />
     </projectFiles>
 
     <issueHandlers>
diff --git a/Exception/UnsupportedPsalmVersion.php b/src/Exception/UnsupportedPsalmVersion.php
similarity index 100%
rename from Exception/UnsupportedPsalmVersion.php
rename to src/Exception/UnsupportedPsalmVersion.php
diff --git a/hooks/TestCaseHandler.php b/src/Hooks/TestCaseHandler.php
similarity index 100%
rename from hooks/TestCaseHandler.php
rename to src/Hooks/TestCaseHandler.php
diff --git a/Plugin.php b/src/Plugin.php
similarity index 72%
rename from Plugin.php
rename to src/Plugin.php
index debc2f1..c538a27 100755
--- a/Plugin.php
+++ b/src/Plugin.php
@@ -15,14 +15,14 @@ class Plugin implements PluginEntryPointInterface
     /** @return void */
     public function __invoke(RegistrationInterface $psalm, SimpleXMLElement $config = null)
     {
-        $psalm->addStubFile(__DIR__ . '/stubs/Assert.php');
+        $psalm->addStubFile(__DIR__ . '/../stubs/Assert.php');
         if ($this->packageVersionIs('phpunit/phpunit', '>=', '7.5')) {
-            $psalm->addStubFile(__DIR__ . '/stubs/Assert_75.php');
+            $psalm->addStubFile(__DIR__ . '/../stubs/Assert_75.php');
         }
-        $psalm->addStubFile(__DIR__ . '/stubs/TestCase.php');
-        $psalm->addStubFile(__DIR__ . '/stubs/MockBuilder.php');
-        $psalm->addStubFile(__DIR__ . '/stubs/InvocationMocker.php');
-        $psalm->addStubFile(__DIR__ . '/stubs/Prophecy.php');
+        $psalm->addStubFile(__DIR__ . '/../stubs/TestCase.php');
+        $psalm->addStubFile(__DIR__ . '/../stubs/MockBuilder.php');
+        $psalm->addStubFile(__DIR__ . '/../stubs/InvocationMocker.php');
+        $psalm->addStubFile(__DIR__ . '/../stubs/Prophecy.php');
 
         class_exists(Hooks\TestCaseHandler::class, true);
         $psalm->registerHooksFromClass(Hooks\TestCaseHandler::class);