Set the axis in the correct place in QUANTUMSTATES()

Closes #954