AB de Villiers to return to one-day action