fix mahdi - torch