MAGETWO-50676: EntityManager introduction

app/code/Magento/Sitemap/Model/ResourceModel/Cms/Page.php

@@ -149,7 +149,7 @@ class Page extends \Magento\Framework\Model\ResourceModel\Db\AbstractDb
         }
 
         if ($isId) {
-            $this->entityManager->load(PageInterface::class, $object, $value);
+            $this->entityManager->load($object, $value, PageInterface::class);
         }
         return $this;
     }

lib/internal/Magento/Framework/EntityManager/Observer/AfterEntitySave.php

@@ -27,6 +27,9 @@ class AfterEntitySave implements ObserverInterface
     {
         $entity = $observer->getEvent()->getEntity();
         if ($entity instanceof AbstractModel) {
+            if (method_exists($entity->getResource(), 'loadAllAttributes')) {
+                $entity->getResource()->loadAllAttributes();
+            }
             $entity->getResource()->afterSave($entity);
             $entity->afterSave();
             $entity->getResource()->addCommitCallback([$entity, 'afterCommitCallback']);