diff --git a/tests/QueryTest.php b/tests/QueryTest.php
index b162848..30a3e0c 100644
--- a/tests/QueryTest.php
+++ b/tests/QueryTest.php
@@ -1025,7 +1025,9 @@ public function testQueryFromCacheAfterDelete()
             ->andWhere('email','==','john@example.com')
     		->resultDocuments();
 
-        $this->assertEquals($id2, current($results)->getId());
+        $r = current($results);
+
+        $this->assertEquals($id2, $r->getId());
 
         $db->flush(true);
     }
@@ -1070,8 +1072,10 @@ public function testQueryFromCacheAfterSave()
             ->andWhere('email','==','john@example.com')
     		->resultDocuments();
 
-        $this->assertEquals($id2, current($results)->getId());
-        $this->assertEquals('John', current($results)->name);
+        $r = current($results);
+
+        $this->assertEquals($id2, $r->getId());
+        $this->assertEquals('John', $r->name);
 
         $db->flush(true);
     }