Maliks Muwatta Book 16, Hadith Number 47.

Yahya related to me from Malik that he heard that A’isha said, “The Messenger of Allah, may Allah bless him and grant him peace, said, ‘No prophet dies until he is given the choice.’ ” She continued, “I heard him say, ‘O Allah, the highest company,’ and I knew that he was going.”