Suggest Treatment For Burning Sensation In My Penis

I had unprotected oral sex with a girl I met at a party 5 days ago. She performed fellatio on me. The very next day I had a slight burning sensation at the tip of my penis. This got much worse the next day and so I sought out a doctor who suggested it was likely a fungal infection of the urethra caused by an infection in the girls mouth. I am booked to have an STD check after two weeks have elapsed just to check if it s something like chlamydia or gonorrhea. The burning sensation is nearly gone after two days of anti fungal medication and I have had no burning while peeling or discharge and my urine is clear. Does the diagnosis sound correct to you? Thanks