صحيح مسلم
From Sahih Muslim (Hadith 2143):
Umar reported Allah's Apostle (ﷺ) as saying:The dead is punished in the grave because of wailing on it
Reference: Muslim 2143