Heroku build fix