Fixing heroku errors (2)