Fix build to deploy to heroku