Joe Burns has cut short his stay at Leicestershire and will return to Australia for treatment on a hip problem.