Fix YARD issues