diff --git a/tests/CollectionExtendedTest.php b/tests/CollectionExtendedTest.php
index <HASH>..<HASH> 100644
--- a/tests/CollectionExtendedTest.php
+++ b/tests/CollectionExtendedTest.php
@@ -607,8 +607,11 @@ class CollectionExtendedTest extends
 
 
         $exampleDocument   = Document::createFromArray(array('someNewAttribute' => 'someNewValue'));
-        $resultingDocument = $collectionHandler->byExample($collection->getId(), $exampleDocument);
-
+        $cursor = $collectionHandler->byExample($collection->getId(), $exampleDocument);
+        $this->assertTrue(
+            $cursor->getCount() == 1,
+            'should return 1.'
+        );
 
         $exampleDocument = Document::createFromArray(array('someOtherAttribute' => 'someOtherValue'));
         $result          = $collectionHandler->removeByExample(
@@ -776,6 +779,11 @@ class CollectionExtendedTest extends
         }
 
         $this->assertTrue(
+            $cursor->getCount() == 2,
+            'should return 2.'
+        );
+
+        $this->assertTrue(
             ($resultingDocument[0]->getKey() == 'test1' && $resultingDocument[0]->firstName == 'Joe'),
             'Document returned did not contain expected data.'
         );
Extended test with getCount() for byExample()