diff --git a/src/Opg/Core/Model/Entity/CaseItem/Page/Page.php b/src/Opg/Core/Model/Entity/CaseItem/Page/Page.php
index 5295f675..b670d385 100644
--- a/src/Opg/Core/Model/Entity/CaseItem/Page/Page.php
+++ b/src/Opg/Core/Model/Entity/CaseItem/Page/Page.php
@@ -29,7 +29,7 @@ class Page implements EntityInterface, \IteratorAggregate
     private $id;
 
     /**
-     * @ORM\ManyToOne(targetEntity = "Opg\Core\Model\Entity\CaseItem\Document\Document", inversedBy = "pages")
+     * @ORM\ManyToOne(targetEntity = "Opg\Core\Model\Entity\Documents\IncomingDocument", inversedBy = "pages")
      * @var
      */
     private $document;
diff --git a/tests/OpgTest/Core/Model/Entity/CaseItem/Epa/EpaTest.php b/tests/OpgTest/Core/Model/Entity/CaseItem/Epa/EpaTest.php
index 878018a1..93d4089c 100644
--- a/tests/OpgTest/Core/Model/Entity/CaseItem/Epa/EpaTest.php
+++ b/tests/OpgTest/Core/Model/Entity/CaseItem/Epa/EpaTest.php
@@ -9,7 +9,6 @@
 
 use Doctrine\Common\Collections\ArrayCollection;
 use Opg\Common\Exception\UnusedException;
-use Opg\Core\Model\Entity\CaseItem\Document\Document;
 use Opg\Core\Model\Entity\CaseItem\Epa\Epa;
 use Opg\Core\Model\Entity\CaseItem\Page\Page;
 use Opg\Common\Model\Entity\DateFormat as OPGDateFormat;
@@ -18,7 +17,6 @@
 use Opg\Core\Model\Entity\CaseActor\Correspondent;
 use Opg\Core\Model\Entity\CaseActor\Donor;
 use Opg\Core\Model\Entity\Documents\IncomingDocument;
-use Opg\Core\Model\Entity\Documents\OutgoingDocument;
 
 /**
  * Epa test case.
diff --git a/tests/OpgTest/Core/Model/Entity/CaseItem/Lpa/LpaTest.php b/tests/OpgTest/Core/Model/Entity/CaseItem/Lpa/LpaTest.php
index f6ea47fd..6d60f3bb 100644
--- a/tests/OpgTest/Core/Model/Entity/CaseItem/Lpa/LpaTest.php
+++ b/tests/OpgTest/Core/Model/Entity/CaseItem/Lpa/LpaTest.php
@@ -3,7 +3,6 @@
 
 use Doctrine\Common\Collections\ArrayCollection;
 use Opg\Common\Exception\UnusedException;
-use Opg\Core\Model\Entity\CaseItem\Document\Document;
 use Opg\Core\Model\Entity\CaseItem\Lpa\Lpa;
 use Opg\Core\Model\Entity\CaseActor\Donor;
 use Opg\Core\Model\Entity\CaseItem\Page\Page;