Update wb_controller.rb