diff --git a/tests/GraphQL/Legacy/AddElementToAreaMutationTest.php b/tests/GraphQL/Legacy/AddElementToAreaMutationTest.php
index e3e23cc1..3cc12dea 100644
--- a/tests/GraphQL/Legacy/AddElementToAreaMutationTest.php
+++ b/tests/GraphQL/Legacy/AddElementToAreaMutationTest.php
@@ -4,8 +4,8 @@
 
 use DNADesign\Elemental\GraphQL\AddElementToAreaMutation;
 use DNADesign\Elemental\Models\ElementalArea;
-use DNADesign\Elemental\Tests\GraphQL\FakeResolveInfo;
 use DNADesign\Elemental\Tests\Src\TestElement;
+use SilverStripe\GraphQL\Tests\Fake\FakeResolveInfo;
 use InvalidArgumentException;
 use SilverStripe\Dev\SapphireTest;
 use SilverStripe\GraphQL\Schema\Schema;
@@ -81,7 +81,7 @@ protected function runMutation($className, $elementalAreaID, $afterElementId = n
     {
         $mutation = new AddElementToAreaMutation();
         $context = ['currentUser' => Security::getCurrentUser()];
-        $resolveInfo = new FakeResolveInfo();
+        $resolveInfo = new FakeResolveInfo([]);
 
         $args = [
             'className' => $className,
diff --git a/tests/GraphQL/Legacy/SortBlockMutationCreatorTest.php b/tests/GraphQL/Legacy/SortBlockMutationCreatorTest.php
index b89478b8..15a62b31 100644
--- a/tests/GraphQL/Legacy/SortBlockMutationCreatorTest.php
+++ b/tests/GraphQL/Legacy/SortBlockMutationCreatorTest.php
@@ -3,8 +3,8 @@
 namespace DNADesign\Elemental\Tests\Legacy\GraphQL;
 
 use DNADesign\Elemental\GraphQL\SortBlockMutationCreator;
-use DNADesign\Elemental\Tests\GraphQL\FakeResolveInfo;
 use DNADesign\Elemental\Tests\Src\TestElement;
+use SilverStripe\GraphQL\Tests\Fake\FakeResolveInfo;
 use SilverStripe\Dev\SapphireTest;
 use SilverStripe\GraphQL\Schema\Schema;
 use SilverStripe\Security\Security;
@@ -56,7 +56,7 @@ protected function runMutation($id, $afterBlockId)
 
         $mutation = new SortBlockMutationCreator();
         $context = ['currentUser' => $member];
-        $resolveInfo = new FakeResolveInfo();
+        $resolveInfo = new FakeResolveInfo([]);
 
         $mutation->resolve(null, [
             'id' => $id,