Was David’s Son Punished for David’s Sin?

{Was David’s Son Punished for David’s Sin?} In a past video, it sounded like Mike said David's child dying was "collateral damage.” How is that not punishment for David's sins?