Fix heroku bugs