Fixed warning on heroku