Try to fix torch