update and also modify the `elementByAxis`