Fixed import for python 3. Tox ftw