The penalty NZ got should have been assessed after they got back up to speed. As it was they were just getting things sorted and getting back up to speed so really no penalty. O had to tack to avoid them and lost more ground then what the penalty on NZ gave them.
+1 Just as you said, clearly visible here on the penalty from the Umpire Booth Replay
