diff --git a/src/Service/UDS/UDSClient.php b/src/Service/UDS/UDSClient.php
index 23f9286..3d3f161 100644
--- a/src/Service/UDS/UDSClient.php
+++ b/src/Service/UDS/UDSClient.php
@@ -142,7 +142,7 @@ class UDSClient
 
     public function getMaxAvailableSeats(int $organizationalUnitId): int
     {
-        $organizationalUnit = $this->entityManager->getRepository(OrganizationalUnit::class)->findOneBy(['migrationId' => $organizationalUnitId]);
+        $organizationalUnit = $this->entityManager->getRepository(OrganizationalUnit::class)->findOneBy(['id' => $organizationalUnitId]);
 
         if (!$organizationalUnit) {
             return 0;