Fix torch

Change-Id: Idf08e2ced55ed0fe2a7150934d42fcffac08d798