Fixed an import line from last commit that broke on python3.