Fixing heroku UI issue