Fix controller for Heroku