Update before_restart.rb