Few points later, you could argue the fielding side still regarded it as in play.
20.1.2 The ball shall be considered to be dead when it is clear to the bowler’s end umpire that the fielding side and both batters at the wicket have ceased to regard it as in play