Update staging.rb to match production.rb