Looking for advice

Since having a baby 2 months ago, Ive started to resent my partner. I find myself having little patience for him and all the things we used to do, such as our way of playing around and joking with eachother. I get pretty snarky and snap at him often. Whenever we go on weekly date nights, its easier to be happy and have fun with him, but we only have limited time to be together without the baby. As soon as the baby is around Im back to primarily focusing on baby, and not wanting to enjoy my partners company. I feel so terrible about this, my partner deserves so much better. Im not sure how to better myself for him. Has anyone else has issues like this? How did you fix yourself?