re-introducing dairy while breastfeeding

So I stopped eating obvious dairy a while ago because my son had bad gas (I still eat hidden dairy aka dairy that's cooked or baked into things). I didn't know if it was just because his digesting system was still maturing or what but gave it a try. He's now 2 months and he's better but I don't if it's because of the dairy or his body is just maturing more. He didn't have any other symptoms of an allergy or anything, so any idea when I can start re-introducing dairy again? I have been researching some some mamas say 4-6 months but their babies had much more severe reactions to dairy, not just gassy. TIA!!