fix heroku issue
minor style tweak