diff --git a/lib/ash_authentication/user_identity/upsert_identity_change.ex b/lib/ash_authentication/user_identity/upsert_identity_change.ex
index 9ced34e..d7115c9 100644
--- a/lib/ash_authentication/user_identity/upsert_identity_change.ex
+++ b/lib/ash_authentication/user_identity/upsert_identity_change.ex
@@ -34,7 +34,7 @@ defmodule AshAuthentication.UserIdentity.UpsertIdentityChange do
       # uid is a convention
       # sub is supposedly from the spec
       # id is from what has been seen from Google
-      |> Map.take(["uid", "sub", "id"])
+      |> Map.take(["uid", "sub", "id", :uid, :sub, :id])
       |> Map.values()
       |> Enum.reject(&is_nil/1)
       |> List.first()