Fixed torch placement.

This is related to #127, but until we know more about the change, I'm not closing the issue.