Could be he felt cold when you undress him. Maybe you can just wipe him him with warm water from the pail first after few weeks then maybe you can slowly bath him in the bathtub.
That's what I have been doing till now. My boy is already 3 months so far is he so excited when come to undress and bath him. He will just keep on splashing water with his leg in the tub.