BUG: fixed issues where passing an axis of 'index' would fail (as its expecting an axis number)