Yah, smaller and younger babies should not stay long in the bath. Older babies tend to play in the water and thus, takes time and it is alright also that they stay longer.