Fix heroku compile problem