Sri Lanka were undoing their good work earlier in the day after a poor batting display on the opening day of the second test against New Zealand at the Basin Reserve.
Having rattled through the last eight New Zealand wickets for just 80 to dismiss the hosts for 221, Sri Lanka were on the run at stumps at 78 for five.
New Zealand's seamers had made better use of favourable bowling conditions with the recalled Doug Bracewell leading the way with two wickets in his opening six-over spell after 14 months out of the test side, and a third in the day's final over to finish with three for 23 in 7.4 overs.
Trent Boult made the initial breakthrough having Dimuth Karunaratne caught at third slip in his fifth over.
Fellow opener Kaushal Silva was unlucky, jamming a Bracewell delivery down only to have it bounce up and land on the top of the stumps.