Yes, you are right... I got confused by the fact that the condition for the `start` (i.e the second) axis is `<= ndim`, not the typical `< ndim`. Is it worth adding a check in the tests.