I guess that I am lucky in some ways. My son, three months old. Loves being in the bath. He is not scared of water and loves to splash.
However, taking him out of the bath is a different matter, often it will set him off screaming, and crying which can last till after he is dried and dressed.
Is he upset about leaving the bath? Or is it likely to be the sensation of being wet but not in the bath that he likes. Are there ways to make getting out of the bath less shocking for him.
I normally pick him out and wrap him in a towel straight away.