Fix bug that was dropping axis names