Update railtie.rb

Updating imports