Sri Lanka have fought their way back into the second test against the Black Caps by claiming three wickets, including Ross Taylor and Kane Williamson, in the opening session on day two.
New Zealand were 298-5 at lunch with Daniel Flynn (6) and debutant Todd Astle (2) at the crease.
Taylor and Williamson took their third wicket partnership through to 262, and the New Zealand tally to 276, before Taylor was trapped LBW for 142 by Rangana Herath.
The New Zealand skipper was attempting to sweep the spinner and was trapped plumb in front.
Herath also dismissed Williamson in similar fashion for 135 after the youngster brought up his third test century earlier in the session.
The score is Williamson's highest in test cricket surpassing his previous best of 131 in India.
Williamson's demise left New Zealand 290-4 and it was soon 291-5 when Kruger van Wyk failed to trouble the scorers. The New Zealand keeper was bowled by Tillakaratne Dilshan after lasting just three balls.