Changing the mutation update resolver to allow for _id instead of id

[full-stacked]

I am using NestJs, GraphQL, TypeORM and MongoDB. When I try to do a simple updateOne like this:

mutation {
  updateOneNotification(input: 
    { 
      id: "6610a12c-237e-4c46-b4c2-ca750029dc2b",
      update: { message: "this has been updated" } 
    }){
    message
  }
}

I received Query Builder is not supported by MongoDB

I've created a base entity with the following fields

  @ObjectIdColumn()
  _id: string;

  @IDField(() => ID)
  @PrimaryGeneratedColumn('uuid')
  id!: string;

I've tried changing the id to the ObjectIdColumn and many other things but I believe the issue is the automatically handled updateOne resolver is trying to update using the id instead of _id. So it would need to be
updateOne({ id })

Is there a built-in solution for this or should I work to override the update resolvers to use my own logic? Any help would be appreciated!