I actually dislike this change, him being distracted thinking about sex while chaeyoon's telling his testimony is so out of character of him ://
Also, I think nobody is surprised just because of the sex, it has to do with the way it was introduced, it came out of nowhere (maybe it's more shocking for us who read the novel ¿)