fixed torch issue