Babylonian Talmud, Tractate Qiddushin

גמ אמר רבי אלעזר ברבי צדוק למה צדיקים נמשלים בעולם הזה לאילן שכולו עומד במקום טהרה ונופו נוטה למקום טומאה נקצץ נופו כולו עומד במקום טהרה כך הקב"ה מביא יסורים על צדיקים בעולם הזה כדי שיירשו העולם הבא שנאמר והיה ראשיתך מצער ואחריתך ישגה מאד ולמה רשעים דומים בעולם הזה לאילן שכולו עומד במקום טומאה ונופו נוטה למקום טהרה נקצץ נופו כולו עומד במקום טומאה כך הקב"ה משפיע להן טובה לרשעים בעולם הזה כדי לטורדן ולהורישן למדריגה התחתונה שנאמר יש דרך ישר לפני איש ואחריתה דרכי מות