Update to fix torch

Change-Id: I7fd69601037a0706351ddb919612a31dc288f5dd