Fix torch error