diff --git a/tests/bundle/Gateway/DoctrineSchemaMetadataTest.php b/tests/bundle/Gateway/DoctrineSchemaMetadataTest.php
index 7bc2ea2..6b8e210 100644
--- a/tests/bundle/Gateway/DoctrineSchemaMetadataTest.php
+++ b/tests/bundle/Gateway/DoctrineSchemaMetadataTest.php
@@ -8,6 +8,7 @@
 
 namespace Ibexa\Tests\Bundle\CorePersistence\Gateway;
 
+use Doctrine\DBAL\Connection;
 use Ibexa\Contracts\CorePersistence\Exception\RuntimeMappingExceptionInterface;
 use Ibexa\Contracts\CorePersistence\Gateway\DoctrineRelationshipInterface;
 use Ibexa\Contracts\CorePersistence\Gateway\DoctrineSchemaMetadata;
@@ -20,7 +21,7 @@ final class DoctrineSchemaMetadataTest extends TestCase
     protected function setUp(): void
     {
         $this->metadata = new DoctrineSchemaMetadata(
-            $this->createMock(\Doctrine\DBAL\Connection::class),
+            $this->createMock(Connection::class),
             'stdClass',
             'std_class_table',
             [
diff --git a/tests/bundle/Gateway/ExpressionVisitorTest.php b/tests/bundle/Gateway/ExpressionVisitorTest.php
index 7e9fc27..889f397 100644
--- a/tests/bundle/Gateway/ExpressionVisitorTest.php
+++ b/tests/bundle/Gateway/ExpressionVisitorTest.php
@@ -189,7 +189,7 @@ public function testFieldFromSubSelectRelationship(): void
         $this->connection
             ->expects(self::once())
             ->method('createQueryBuilder')
-            ->willReturnCallback(fn () => new QueryBuilder($this->connection));
+            ->willReturnCallback(fn (): QueryBuilder => new QueryBuilder($this->connection));
 
         /** @var class-string $relationshipClass pretend it's a class-string */
         $relationshipClass = 'relationship_class';