diff --git a/lib/src/main/java/graphql/nadel/engine/util/GraphQLUtil.kt b/lib/src/main/java/graphql/nadel/engine/util/GraphQLUtil.kt
index 1e9a6b9f6..c0c0358f1 100644
--- a/lib/src/main/java/graphql/nadel/engine/util/GraphQLUtil.kt
+++ b/lib/src/main/java/graphql/nadel/engine/util/GraphQLUtil.kt
@@ -44,7 +44,10 @@ import graphql.nadel.instrumentation.parameters.NadelInstrumentationExecuteOpera
 import graphql.nadel.util.ErrorUtil.createGraphQLErrorsFromRawErrors
 import graphql.normalized.ExecutableNormalizedField
 import graphql.normalized.ExecutableNormalizedOperation
+import graphql.normalized.ExecutableNormalizedOperationToAstCompiler
+import graphql.normalized.ExecutableNormalizedOperationToAstCompiler.CompilerResult
 import graphql.normalized.NormalizedInputValue
+import graphql.normalized.VariablePredicate
 import graphql.schema.FieldCoordinates
 import graphql.schema.GraphQLCodeRegistry
 import graphql.schema.GraphQLFieldDefinition
@@ -556,7 +559,7 @@ fun compileToDocument(
     operationName: String?,
     topLevelFields: List<ExecutableNormalizedField>,
     variablePredicate: VariablePredicate?,
-): ExecutableNormalizedOperationToAstCompiler.CompilerResult {
+): CompilerResult {
     return ExecutableNormalizedOperationToAstCompiler.compileToDocument(
         schema,
         operationKind,
diff --git a/lib/src/test/kotlin/graphql/nadel/instrumentation/ChainedNadelInstrumentationTest.kt b/lib/src/test/kotlin/graphql/nadel/instrumentation/ChainedNadelInstrumentationTest.kt
index 773a0e4d4..2647cc87a 100644
--- a/lib/src/test/kotlin/graphql/nadel/instrumentation/ChainedNadelInstrumentationTest.kt
+++ b/lib/src/test/kotlin/graphql/nadel/instrumentation/ChainedNadelInstrumentationTest.kt
@@ -122,7 +122,7 @@ class ChainedNadelInstrumentationTest : DescribeSpec({
                 } returns chainedState
 
                 every {
-                    params.copy(any(), any(), any(), any(), any())
+                    params.copy(any(), any(), any(), any(), any(), any())
                 } answers { copyCall ->
                     val newState = copyCall.invocation.args.singleOfType<InstrumentationState>()
                     mock { newParams ->
@@ -372,7 +372,7 @@ class ChainedNadelInstrumentationTest : DescribeSpec({
             val paramsCopy = mock<NadelInstrumentationTimingParameters>()
             val params = mock<NadelInstrumentationTimingParameters> { params ->
                 every { params.getInstrumentationState<InstrumentationState>() } returns chainedState
-                every { params.copy(any(), any(), any(), any(), any()) } returns paramsCopy
+                every { params.copy(any(), any(), any(), any(), any(), any()) } returns paramsCopy
             }
 
             // when