Fix import for py3k