He added 47 for the fifth wicket with skipper Mahela Jayawardene (24), the highest stand of the innings, after four wickets had fallen for 44 runs.

"We are looking forward to the Test series now as we haven't played Test cricket for some time now. We'll try to be as consistent as possible," said Jayawardene.

Southee did the early damage when he bowled opener Dinesh Chandimal for no score and then had Lahiru Thirimanne caught behind to reduce Sri Lanka to 33-2.

Thirimanne, promoted in the batting order, contributed a 15-ball 17 with three boundaries.

Boult dismissed Angelo Mathews before Milne got his maiden one-day wicket when he had Kumar Sangakkara caught behind for five.

Sri Lanka made two changes from the side that played the last match, bringing in paceman Shaminda Eranga and debutant spinner Akila Dananjaya in place of Nuwan Kulasekara and Rangana Herath. New Zealand retained the same side.

Sri Lanka and New Zealand will play a two-Test series starting in Galle on Saturday.