Fixed bug in axis assignments