diff --git a/Tests/Transport/PostgreSqlConnectionTest.php b/Tests/Transport/PostgreSqlConnectionTest.php
index e8e00d9..c542908 100644
--- a/Tests/Transport/PostgreSqlConnectionTest.php
+++ b/Tests/Transport/PostgreSqlConnectionTest.php
@@ -11,8 +11,8 @@
 
 namespace Symfony\Component\Messenger\Bridge\Doctrine\Tests\Transport;
 
-use Doctrine\DBAL\Cache\ArrayResult;
 use Doctrine\DBAL\Cache\ArrayStatement;
+use Doctrine\DBAL\Driver\Result as DriverResult;
 use Doctrine\DBAL\Platforms\PostgreSQLPlatform;
 use Doctrine\DBAL\Query\QueryBuilder;
 use Doctrine\DBAL\Result;
@@ -95,10 +95,13 @@ public function countNotifyCalls()
                 ->method('getNativeConnection')
                 ->willReturn($wrappedConnection);
 
+            $driverResult = $this->createMock(DriverResult::class);
+            $driverResult->method('fetchAssociative')
+                ->willReturn(false);
             $driverConnection
                 ->expects(self::any())
                 ->method('executeQuery')
-                ->willReturn(new Result(new ArrayResult([]), $driverConnection));
+                ->willReturn(new Result($driverResult, $driverConnection));
         }
         $connection = new PostgreSqlConnection(['table_name' => 'queue_table'], $driverConnection);