Add `from astropy import units as u` here so that this code example works standalone like all other code examples.