fix enable torch, thanks rascarlo