diff --git a/src/Testing/EntityManagerMock.php b/src/Testing/EntityManagerMock.php
index f165c0d..c7cb107 100644
--- a/src/Testing/EntityManagerMock.php
+++ b/src/Testing/EntityManagerMock.php
@@ -2,6 +2,7 @@
 
 namespace ORM\Testing;
 
+use Mockery as m;
 use ORM\Entity;
 use ORM\EntityFetcher;
 use ORM\EntityManager;
@@ -56,7 +57,7 @@ public function retrieve($class, array $primaryKey)
      *
      * @param $class
      * @param Entity ...$entities
-     * @return Result
+     * @return Result|m\MockInterface
      * @codeCoverageIgnore proxy method
      */
     public function addResult($class, Entity ...$entities)
diff --git a/src/Testing/MocksEntityManager.php b/src/Testing/MocksEntityManager.php
index 4518ef2..cc4e42b 100644
--- a/src/Testing/MocksEntityManager.php
+++ b/src/Testing/MocksEntityManager.php
@@ -129,7 +129,7 @@ function ($str) {
      *
      * @param string $class The class of an Entity
      * @param Entity ...$entities The entities that will be returned
-     * @return EntityFetcherMock\Result
+     * @return EntityFetcherMock\Result|m\MockInterface
      * @codeCoverageIgnore trivial code
      */
     public function ormAddResult($class, Entity ...$entities)