Update to fix Heroku