Fix YARD coverage