Fix Torch

Change-Id: I4bee6446b44e2bf9e1165bc7f1884ff7fbd7ef05