Fix import for Python 3.3 support