Should we just rename the parameter `axes`?