Возможно, малыша просто перевозбуждает купание. Попробуйте купать его днем или утром (только не перед прогулкой), а вечером вместо купания погуляйте на улице.