Revert to torch!