yuga: Fix torch

Change-Id: Ia384339ec2d69ead77472071df5120f1c066a564