Swap dimensions used by `reshape`

Betrayed by my IDL background. A good catch by @mcflugen.