Add a unit argument to Uuseg.custom.