Maliks Muwatta Book 3, Hadith Number 69.
Section : Completing what is Recalled When Uncertain how much has been Prayed.
Yahya related to me from Malik from Nafi that Abdullah ibn Umar, when questioned about forgetfulness in the prayer, said, “If you think that you have forgotten part of the prayer, then pray it.”