Chapter 8 · The Imperishable Absolute · Krishna said
Bhagavad Gita 8.6 — As you think, so you go
यं यं वापि स्मरन्भावं त्यजत्यन्ते कलेवरम्। तं तमेवैति कौन्तेय सदा तद्भावभावितः ॥
yaṃ yaṃ vāpi smaranbhāvaṃ tyajatyante kalevaram | taṃ tamevaiti kaunteya sadā tadbhāvabhāvitaḥ ||
Meaning
Whatever state of being one remembers when leaving the body at the end, Arjuna, that one reaches — because one has always been absorbed in it.