slight change to the import in __init__.py to preserve py3 compatibility