Fix compile issue for heroku