Still fixing heroku error in controller.