Fix YARD docs