Did I do the right thing?

I love my husband, but it’s just not fair to me. Every so often my husband goes through a difficult time. He gets so stressed out and overwhelms himself that he shuts down and shuts everyone out. Including me. He’s currently going through this again and I know nothing. Idk why he’s like this or what triggered it because he won’t talk to me. Last time he got this way, he told me he didn’t want to be with me anymore. That he wasn’t happy and wanted a divorce. We separated for about a month a half before we got back together under the conditions that we had to go to marriage counseling. And it went pretty well. It helped us a lot. But now that he’s acting this way again I’m worried that it might be the same thing again. I’m really not sure but I had to set a boundary. It’s not fair to me that he shuts me away when he acts this way. I told him I wasn’t tolerating this anymore. That I would give him his space for a few days and then he has to talk to me. I know it sounds like I’m forcing him now but I just don’t think it’s fair for me to put up with this anymore. We’re married. We’re suppose to be there for each other. What’s the point of being with someone if every time you get stressed out, you push them away? I believe couples are supposed to evolve together and always be there for each other. Plus, I’m currently 8 months pregnant. I don’t need the stress I go through when he acts this way. Idk. Idk what to say or think anymore.