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.