Auto generated `reset_password_with_token` not loading user

update :reset_password_with_token do
  argument :reset_token, :string do
    allow_nil? false
    sensitive? true
  end

  argument :password, :string do
    description "The proposed password for the user, in plain text."
    allow_nil? false
    constraints min_length: 8
    sensitive? true
  end

  argument :password_confirmation, :string do
    description "The proposed password for the user (again), in plain text."
    allow_nil? false
    sensitive? true
  end

  # validates the provided reset token
  validate AshAuthentication.Strategy.Password.ResetTokenValidation

  # validates that the password matches the confirmation
  validate AshAuthentication.Strategy.Password.PasswordConfirmationValidation

  # Hashes the provided password
  change AshAuthentication.Strategy.Password.HashPasswordChange

  # Generates an authentication token for the user
  change AshAuthentication.GenerateTokenChange
end

mutations do
update :reset_password_with_token, :reset_password_with_token do
  identity :unique_email
end

action :request_password_reset_token, :request_password_reset_token

mutation RequestPasswordResetToken {
  requestPasswordResetToken(input: { email: "[email protected]" })
}

mutation ResetPasswordWithToken {
  resetPasswordWithToken(
    input: {
      password: "123456789"
      passwordConfirmation: "123456789"
      resetToken: "token_from_previous_step"
    }
    email: "[email protected]"
  ) {
    result {
      id
      email
    }
  }
}