Australia working to beat New Zealand swing