diff --git a/src/Psalm/Plugin/PluginEntryPointInterface.php b/src/Psalm/Plugin/PluginEntryPointInterface.php
index abeb0d1733b..511888d64aa 100644
--- a/src/Psalm/Plugin/PluginEntryPointInterface.php
+++ b/src/Psalm/Plugin/PluginEntryPointInterface.php
@@ -4,9 +4,10 @@
 
 namespace Psalm\Plugin;
 
+use Psalm\PluginRegistrationSocket;
 use SimpleXMLElement;
 
 interface PluginEntryPointInterface extends PluginInterface
 {
-    public function __invoke(RegistrationInterface $registration, ?SimpleXMLElement $config = null): void;
+    public function __invoke(PluginRegistrationSocket $registration, ?SimpleXMLElement $config = null): void;
 }