Fixed import

Hopefully fixed some import problems with BeautifulSoup. Coming back to
this project, the script wasn't working and changing this line helped.