diff --git a/src/Bundle/ChillMainBundle/Tests/Export/ExportGeneratorTest.php b/src/Bundle/ChillMainBundle/Tests/Export/ExportGeneratorTest.php
index 2faa49c22..e01dc6566 100644
--- a/src/Bundle/ChillMainBundle/Tests/Export/ExportGeneratorTest.php
+++ b/src/Bundle/ChillMainBundle/Tests/Export/ExportGeneratorTest.php
@@ -324,6 +324,10 @@ class ExportGeneratorTest extends TestCase
         $formatter = $this->prophesize(FormatterInterface::class);
 
         $query = $this->prophesize(QueryBuilder::class);
+        $query->getDQL()->willReturn('dummy');
+        $dqlQuery = $this->prophesize(Query::class);
+        $dqlQuery->getSQL()->willReturn('dummy');
+        $query->getQuery()->willReturn($dqlQuery->reveal());
 
         // required methods
         $export->initiateQuery(