كتاب الطلاق
18  
The Book of Divorce
(8)
Chapter: The 'Iddah of a woman whose husband had died, and the like, ends when she gives birth.
(8)
باب انْقِضَاءِ عِدَّةِ الْمُتَوَفَّى عَنْهَا زَوْجُهَا وَغَيْرِهَا بِوَضْعِ الْحَمْلِ ‏‏
Sahih Muslim 1485 b

This hadith has been narated with the same chain of transrmitters except with a small change of words (and that is):

They sent him to Umm Salama, but no mention was made of Kuraib.
وَحَدَّثَنَاهُ مُحَمَّدُ بْنُ رُمْحٍ، أَخْبَرَنَا اللَّيْثُ، ح وَحَدَّثَنَاهُ أَبُو بَكْرِ بْنُ أَبِي شَيْبَةَ، وَعَمْرٌو، النَّاقِدُ قَالاَ حَدَّثَنَا يَزِيدُ بْنُ هَارُونَ، كِلاَهُمَا عَنْ يَحْيَى بْنِ سَعِيدٍ، بِهَذَا الإِسْنَادِ ‏.‏ غَيْرَ أَنَّقَالَ فِي حَدِيثِهِ فَأَرْسَلُوا إِلَى أُمِّ سَلَمَةَ وَلَمْ يُسَمِّ كُرَيْبًا ‏.‏
Reference : Sahih Muslim 1485b
In-book reference : Book 18, Hadith 72
USC-MSA web (English) reference : Book 9, Hadith 3538
  (deprecated numbering scheme)