oops, fix axis normalization