Fix a bug for heroku deployment