Reference for Bekhorot 68:28

כוון מלאכתו במועד ומת לא קנסו בנו אחריו משום דלא עבדא לאיסורא

from the first plague and as regards the second, we wait seven days for him.

Jastrow