Fixed error for use in Heroku