Small change to fix gem