Fixed import error on Python 3.4