Dustin Johnson knew there was a possible stroke taken away