Fix torch

Change-Id: Iaa8bc082b59bcae1fc6c3bde566e7374e7f76754