diff --git a/tests/Integration/Database/QueryBuilderTest.php b/tests/Integration/Database/QueryBuilderTest.php
index 675f30561fae..66a65cbdaa43 100644
--- a/tests/Integration/Database/QueryBuilderTest.php
+++ b/tests/Integration/Database/QueryBuilderTest.php
@@ -36,7 +36,7 @@ public function testSole()
     {
         $expected = ['id' => '1', 'title' => 'Foo Post'];
 
-        $this->assertEquals(1, DB::table('posts')->where('title', 'Foo Post')->sole()->id);
+        $this->assertSame($expected, (array) DB::table('posts')->where('title', 'Foo Post')->select('id', 'title')->sole());
     }
 
     public function testSoleFailsForMultipleRecords()