Fix heroku loading