Update for ruby 2