Fix an import for Python 2.7
Merge pull request #1 from apolkosnik/patch-1

Fix an import for Python 2.7