Boys and guilt over shaving head

Hi all, reaching out for some support/community.

Last night (1/6) found 10-12 nits and 2 bugs in my son’s hair after we got a message from school about lice over the weekend. He has semi long hair so we shaved it, did treatment and comb. Got everything out.

The worst part is the guilt over shaving my 4YOs head. He was upset but then happy after once I gave him a sucker.

I know he’ll be fine and it will grow back but it was pretty traumatizing for me and him during the shaving and once he saw himself. We kept telling him how nice he looks and how much he looks like his big kid friends.

Anyone experience this or made this decision?