diff --git a/Model/DonationManagement.php b/Model/DonationManagement.php
index 05910ae..f84a5ab 100644
--- a/Model/DonationManagement.php
+++ b/Model/DonationManagement.php
@@ -89,7 +89,7 @@ public function getByOrder(Order $order): DonationInterface
 
         $this->donationResource->load($entity, $orderId, DonationInterface::ORDER_ID);
 
-        if (!$entity->getEntityId()) {
+        if (!$entity->getDonationId()) {
             throw new NoSuchEntityException(__("The requested donation does not exist."));
         }
 
diff --git a/Model/DonationRepository.php b/Model/DonationRepository.php
index b63d5e1..f2e7603 100644
--- a/Model/DonationRepository.php
+++ b/Model/DonationRepository.php
@@ -63,7 +63,7 @@ public function get(int $id): DonationInterface
         }
         $entity = $this->donationFactory->create();
         $this->donationResource->load($entity, $id);
-        if (!$entity->getEntityId()) {
+        if (!$entity->getDonationId()) {
             throw new NoSuchEntityException(__("The requested donation does not exist."));
         }
         return $entity;