Add back Torch