Revert "Torch fix"

This reverts commit 86ff90da6fc5db6dc40370769d45de476d74087d.