Fixing error for heroku deploy