diff --git a/src/Test/EntityManagerMockFactoryTrait.php b/src/Test/EntityManagerMockFactoryTrait.php
index c0fe037a..175b81de 100644
--- a/src/Test/EntityManagerMockFactoryTrait.php
+++ b/src/Test/EntityManagerMockFactoryTrait.php
@@ -32,6 +32,8 @@ final protected function createEntityManagerMock(\Closure $qbCallback, array $fi
     {
         $query = $this->createMock(AbstractQuery::class);
         $query->method('execute')->willReturn(true);
+        $query->method('getResult')->willReturn([]);
+        $query->method('getOneOrNullResult')->willReturn([]);
 
         $qb = $this->createMock(QueryBuilder::class);