Revert vectorTH to it's old implementation, so that it works with the new tfp