Yeah, that sounds pretty normal I can't remember when AF returned for me but I do recall that it was pretty early on and I BF'd for 6 months, I had AF at least twice. For me, it was irregular and a bit lighter than usual, I wasn't worried about it until I stopped BFing and it became regular again, but very heavy and painful (ended up going through a bunch of testing for PCOS but didn't get a lot of support to find out what was happening so gave up and lived with it). It's back to normal now, FWIW

AF can return while BFing at any time, the general consensus is that if you go more than 4 hours without a feed, it will probably come back - I think mine returned because DD2 cut herself down to only one night feed, so I was going for about 6 hours through the night without feeding. It shouldn't affect your ability to BF (although bubs can sometimes get a bit fussy at the boob if AF is visiting, the hormone changes have something to do with it), just be more careful with contraception and don't panic if it comes and goes in really strange patterns for a while!