Revert changes to base.py