Fix small bug which appears on switching axes