Remove unnecessary test axis