Trying to fix bug on heroku