Fix torch

Change-Id: I172800cf0a7e6bd941ab4e8eb375746d5c745afc