fixed a bug on heroku