Fix gem load issue