Krishna,Geeta,Chapter 8, Verse 6.

Whatever state of being one remembers when he quits his body, that state he will attain without fail.