diff --git a/tests/Doctrine/Tests/ORM/Functional/PersistentCollectionCriteriaTest.php b/tests/Doctrine/Tests/ORM/Functional/PersistentCollectionCriteriaTest.php
index c2f5e76d2..178e51fc1 100644
--- a/tests/Doctrine/Tests/ORM/Functional/PersistentCollectionCriteriaTest.php
+++ b/tests/Doctrine/Tests/ORM/Functional/PersistentCollectionCriteriaTest.php
@@ -74,8 +74,9 @@ class PersistentCollectionCriteriaTest extends \Doctrine\Tests\OrmFunctionalTest
         $this->loadFixture();
         $repository = $this->_em->getRepository('Doctrine\Tests\Models\Company\CompanyOrganization');
 
-        $organization = $repository->find(1);
-        $events       = $organization->events->matching(new Criteria());
+        $organizations = $repository->findAll();
+        $organization  = $organizations[0];
+        $events        = $organization->events->matching(new Criteria());
 
         $this->assertInstanceOf('Doctrine\ORM\LazyCriteriaCollection', $events);
         $this->assertFalse($events->isInitialized());