diff --git a/lib/utils/transform-schema.util.ts b/lib/utils/transform-schema.util.ts
index 593b2f8ee..24b8f30df 100644
--- a/lib/utils/transform-schema.util.ts
+++ b/lib/utils/transform-schema.util.ts
@@ -2,7 +2,7 @@
 // The changed lines are 31-40 and 85-87 and the original file can be found here:
 // https://github.com/apollographql/apollo-tooling/blob/master/packages/apollo-graphql/src/schema/transformSchema.ts
 
-import { GraphQLReferenceResolver } from '@apollo/federation/dist/types';
+import '@apollo/subgraph/dist/schemaExtensions';
 import {
   GraphQLFieldConfigArgumentMap,
   GraphQLFieldConfigMap,
@@ -27,17 +27,6 @@ import {
   isUnionType,
 } from 'graphql';
 
-// Definitions taken from here: https://github.com/apollographql/apollo-server/blob/main/packages/apollo-federation/src/types.ts#L62
-declare module 'graphql/type/definition' {
-  interface GraphQLObjectType {
-    resolveReference?: GraphQLReferenceResolver<any>;
-  }
-
-  interface GraphQLObjectTypeConfig<TSource, TContext> {
-    resolveReference?: GraphQLReferenceResolver<TContext>;
-  }
-}
-
 type TypeTransformer = (
   type: GraphQLNamedType,
 ) => GraphQLNamedType | null | undefined;