Improving fast-particle confinement in quasi-axisymmetric stellarator optimization