Also, all of the other cases I tested worked fine when ```axis=None```, this was the only case where that didn't work as expected.