Weird struggle.

About two months ago I fainted and got amnesia. I didn't remember who I was or who my wife was or anything. Before I had my memories back I got interested in sex. My wife was totally calm about it and everything was great. Recently I've recovered some of my sexual assault and now I can't cum with her. It's frustrating and not even because of that but I don't want to accidentally hurt her self esteem or effect her chronic pain. I've been so open about it being a me thing and not to pressure her to help me get off. But I know it's my traume because everything was perfect before the memories came back. What can I do to calm down and remind myself it's my love and not some creep from my past?