Missing importing in base.py