Fix torch

Change-Id: I24ff02998a79b9930b92e483a49d1ab9b30dc52c