New Zealand may remove all virus restriction next week