England to punish Black Caps after narrow escape in Dunedin