After returning from a two-day visit to Beijing, the federal treasurer says work was still ongoing to lift a trade ban on Australian lobster.