- if ($this->entityRepo->hasUserGotPageDraft($page, $this->currentUser->id)) {
- $draft = $this->entityRepo->getUserPageDraft($page, $this->currentUser->id);
- $page->name = $draft->name;
- $page->html = $draft->html;
- $page->markdown = $draft->markdown;
+ $userPageDraft = $this->pageRepo->getUserPageDraft($page, $this->currentUser->id);
+ if ($userPageDraft !== null) {
+ $page->name = $userPageDraft->name;
+ $page->html = $userPageDraft->html;
+ $page->markdown = $userPageDraft->markdown;