Fixing bug with heroku push