Bhagwad Gita
Chapter 8, Verse 6
यं यं वापि स्मरन्भावं त्यजत्यन्ते कलेवरम्।
तं तमेवैति कौन्तेय सदा तद्भावभावितः।।8.6।।
Whatever state of being one remembers when he quits his body, that state he will attain without fail.