Bhagavad Gita, Chapter 8, Verse 6

Remembering whatever object, at the end, he leaves the body, that alone is reached by him, O son of Kunti, because of his constant thought of that object.

Whatever's genuinely on someone's mind at that final moment is described as carrying real, lasting weight.

This extends the previous verse into a broader real principle. It isn't only about remembering God specifically, whatever consumes someone's attention in that final moment is described as shaping what comes after.