diff --git a/core/src/test/scala/sangria/federation/v1/FederationSpec.scala b/core/src/test/scala/sangria/federation/v1/FederationSpec.scala
index e1970ea..f572215 100644
--- a/core/src/test/scala/sangria/federation/v1/FederationSpec.scala
+++ b/core/src/test/scala/sangria/federation/v1/FederationSpec.scala
@@ -62,7 +62,6 @@ class FederationSpec extends AsyncFreeSpec {
 
             directive @key(fields: _FieldSet!) repeatable on OBJECT | INTERFACE
           """)
-          .extend(Document(Vector(_FieldSet.Type.toAst)))
 
         schema should beCompatibleWith(expectedSubGraphSchema)
       }
@@ -134,7 +133,6 @@ class FederationSpec extends AsyncFreeSpec {
 
             directive @key(fields: _FieldSet!) repeatable on OBJECT | INTERFACE
           """)
-          .extend(Document(Vector(_FieldSet.Type.toAst)))
 
         schema should beCompatibleWith(expectedSubGraphSchema)
       }
diff --git a/core/src/test/scala/sangria/federation/v2/FederationSpec.scala b/core/src/test/scala/sangria/federation/v2/FederationSpec.scala
index 5b45855..130cba9 100644
--- a/core/src/test/scala/sangria/federation/v2/FederationSpec.scala
+++ b/core/src/test/scala/sangria/federation/v2/FederationSpec.scala
@@ -87,8 +87,6 @@ class FederationSpec extends AsyncFreeSpec {
 
             directive @tag(name: String!) repeatable on FIELD_DEFINITION | INTERFACE | OBJECT | UNION | ARGUMENT_DEFINITION | SCALAR | ENUM | ENUM_VALUE | INPUT_OBJECT | INPUT_FIELD_DEFINITION
           """)
-          .extend(Document(
-            Vector(_FieldSet.Type.toAst, Link__Import.Type.toAst, Link__Purpose.Type.toAst)))
 
         withClue(SchemaRenderer.renderSchema(schema)) {
           schema should beCompatibleWith(expectedSubGraphSchema)
@@ -184,8 +182,6 @@ class FederationSpec extends AsyncFreeSpec {
 
             directive @tag(name: String!) repeatable on FIELD_DEFINITION | INTERFACE | OBJECT | UNION | ARGUMENT_DEFINITION | SCALAR | ENUM | ENUM_VALUE | INPUT_OBJECT | INPUT_FIELD_DEFINITION
           """)
-          .extend(Document(
-            Vector(_FieldSet.Type.toAst, Link__Import.Type.toAst, Link__Purpose.Type.toAst)))
 
         schema should beCompatibleWith(expectedSubGraphSchema)
       }
diff --git a/project/Dependencies.scala b/project/Dependencies.scala
index 3766e69..24e3425 100644
--- a/project/Dependencies.scala
+++ b/project/Dependencies.scala
@@ -7,7 +7,7 @@ object Dependencies {
     val http4s = "1.0.0-M30"
   }
 
-  val sangria = "org.sangria-graphql" %% "sangria" % "4.0.2"
+  val sangria = "org.sangria-graphql" %% "sangria" % "4.1.0"
   val sangriaCirce = "org.sangria-graphql" %% "sangria-circe" % "1.3.2"
   val logbackClassic = "ch.qos.logback" % "logback-classic" % "1.5.6"