Fix types in YARD docs