[Python 3] Fixes another import.