Fix heroku problem