Add missing import to pull.py