Hacky solution to run in both local and heroku