Babylonian Talmud, Tractate Shabbat

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