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