slight fix to controller.rb