Fix torch2