Fixed import incompatible with Python 3.2