Fix swap_axes under PyQT

Closes #349
Merge pull request #357 from ChrisBeaumont/swap_axes

Fix swap_axes under PyQT