Fix Torch

Change-Id: Ie7fd04a0ba7ed09fd1478670119374d24c997b21