Fix axis updating