Debug Heisenberg model

There is some misbehaviour leading to the first dimension of the spin vectors being updated differently than the other two