Update the code to reflect Gemfile update