Baby milk recommendation

Good milk for baby with diarrhoea . Been having diarrhoea going 2 weeks now, went to hospital dr say could be the milk may need to change . Currently taking similac HA total comfort . Can help to suggest which milk best for baby 6 mnth old ?