Fix YARD warning