Fix Heroku compile error