Fix gem version issue