I would either bath together when possible..

or make the bath very shallow and just let him spalsh about on his own and then you only need to sit there next to him