Fix import when using python3.3