Minor tweak in axpy