Wawrinka upped his game, though, and won the second set with a solitary break midway through and then put himself in control by taking Edmund's serve in the opening game of the deciding set and then again to go 5-2 in front.