This simple trick can help - especially for babies under 12 months old.
Try This:

Warm a small towel with warm water then gently lay it over your baby's tummy while they're in the bath.

This helps them feel safe, warm, and secure much like the sensation of being in the womb. It can reduce startle reflex and ease fussiness.

Perfect for newborns and still comforting for babies up to 1 year old.