Ballack close to return