Thank you for your query
It just seems that the normal flora i.e the normal good bacteria in your son's stomach have been upset during the episode of diarrhea
. I don't think you need to worry too much. It should resolve with time. In the meanwhile you can add food items which have normal good bacteria like curd and curd products. Probiotics available in market also help.
God bless you