minor change

now uses the correct axis