Updated torch/flash fix