I think ``axis=None`` is an assertion error