I think we have this in `base.py`