fixed run problem on heroku
added missing import