Set shape and axes on aux data correctly