Fix Torch for DU