diff --git a/app/Models/CompanyModule.php b/app/Models/CompanyModule.php
index e518f2b4..65e0a259 100644
--- a/app/Models/CompanyModule.php
+++ b/app/Models/CompanyModule.php
@@ -113,9 +113,8 @@ class CompanyModule extends AbstractModel implements Addressable, Documentable,
 
     public function getMarking()
     {
-        $connection = $this->invites->first();
-        dd($connection);
-        return $connection ? $connection->pivot->invitee_reference : '';
+        $connection = $this->connections()->first();
+        return $connection ? $connection->invitee_reference : '';
     }
 
     /**