New Zealand head coach Gary Stead will step away from white-ball formats and has a few weeks to decide if he wishes to reapply for the Test role.