Fixed broken heroku build