Treasurer Jim Chalmers has drawn clear election battle lines over one scheme as he prepares to hand down his fourth budget.