re-fix torch

Change-Id: I232ba58a6da6bda0b2b6fa29a9a938434a9d0b30