fixed minor issue with base.py