diff --git a/src/generator/dmmf/transform.ts b/src/generator/dmmf/transform.ts
index f88d78d41..fe2b4397e 100644
--- a/src/generator/dmmf/transform.ts
+++ b/src/generator/dmmf/transform.ts
@@ -396,10 +396,10 @@ function transformMapping(
 
 function selectInputTypeFromTypes(dmmfDocument: DmmfDocument) {
   return (
-    inputTypes: PrismaDMMF.SchemaArgInputType[],
+    inputTypes: PrismaDMMF.TypeRef<PrismaDMMF.FieldLocation>[],
   ): DMMF.SchemaArgInputType => {
     const { useUncheckedScalarInputs, useSimpleInputs } = dmmfDocument.options;
-    let possibleInputTypes: PrismaDMMF.SchemaArgInputType[];
+    let possibleInputTypes: PrismaDMMF.TypeRef<PrismaDMMF.FieldLocation>[];
     possibleInputTypes = inputTypes.filter(
       it =>
         it.location === "inputObjectTypes" &&